diff --git a/invokeai/backend/model_manager/configs/controlnet.py b/invokeai/backend/model_manager/configs/controlnet.py index 630e81fd243..b3b84cabba4 100644 --- a/invokeai/backend/model_manager/configs/controlnet.py +++ b/invokeai/backend/model_manager/configs/controlnet.py @@ -88,7 +88,9 @@ def from_model_on_disk(cls, mod: ModelOnDisk, override_fields: dict[str, Any]) - cls._validate_base(mod) - return cls(**override_fields) + repo_variant = {"repo_variant": override_fields.get("repo_variant", cls._get_repo_variant_or_raise(mod))} + args = override_fields | repo_variant + return cls(**args) @classmethod def _validate_base(cls, mod: ModelOnDisk) -> None: