Conversation
|
@liujij please read the following Contributor License Agreement(CLA). If you agree with the CLA, please reply with the following information.
Contributor License AgreementContribution License AgreementThis Contribution License Agreement (“Agreement”) is agreed to by the party signing below (“You”),
|
| @@ -0,0 +1,169 @@ | |||
| # | |||
Check warning
Code scanning / lintrunner
RUFF-FORMAT/format Warning
| # SPDX-License-Identifier: MIT | ||
| # | ||
|
|
||
| """Olive Pass for Vitis NPU Stable Diffusion submodel generation (UNet / VAE decoder). |
Check warning
Code scanning / lintrunner
RUFF/D205 Warning
| return _SD_CONFIG_REGISTRY | ||
|
|
||
|
|
||
| def _build_fixed_shapes(dim_param: Optional[list], dim_value: Optional[list]) -> Optional[list[str]]: |
Check warning
Code scanning / lintrunner
RUFF/UP007 Warning
| return _SD_CONFIG_REGISTRY | ||
|
|
||
|
|
||
| def _build_fixed_shapes(dim_param: Optional[list], dim_value: Optional[list]) -> Optional[list[str]]: |
Check warning
Code scanning / lintrunner
RUFF/UP007 Warning
| return _SD_CONFIG_REGISTRY | ||
|
|
||
|
|
||
| def _build_fixed_shapes(dim_param: Optional[list], dim_value: Optional[list]) -> Optional[list[str]]: |
Check warning
Code scanning / lintrunner
RUFF/UP007 Warning
|
|
||
|
|
||
| class VitisGenerateModelSD(Pass): | ||
| """Generate Vitis NPU-ready SD submodel (unet or vae_decoder) from ONNX input. |
Check warning
Code scanning / lintrunner
RUFF/D205 Warning
|
|
||
| @classmethod | ||
| def _default_config(cls, accelerator_spec): | ||
| registry = _get_sd_registry() |
Check warning
Code scanning / lintrunner
PYLINT/W0612 Warning
|
|
||
| @classmethod | ||
| def _default_config(cls, accelerator_spec): | ||
| registry = _get_sd_registry() |
Check warning
Code scanning / lintrunner
RUFF/F841 Warning
Describe your changes
Checklist before requesting a review
lintrunner -a(Optional) Issue link