diff --git a/iopaint/model_info.py b/iopaint/model_info.py index d6a5b8d..a053534 100644 --- a/iopaint/model_info.py +++ b/iopaint/model_info.py @@ -58,7 +58,7 @@ class ModelInfo(BaseModel): ModelType.DIFFUSERS_SDXL, ModelType.DIFFUSERS_SD_INPAINT, ModelType.DIFFUSERS_SDXL_INPAINT, - ] + ] or self.name in [POWERPAINT_NAME] @computed_field @property