Skip to content

Commit 8d6e005

Browse files
DustyShoelstein
andauthored
Fix to enable loading fp16 repo variant ControlNets (#8643)
* Fix ControlNet repo variant detection for fp16 weights * Remove ControlNet diffusers fp16 regression test * Update invokeai/backend/model_manager/configs/controlnet.py Co-authored-by: Lincoln Stein <lincoln.stein@gmail.com> * style: ruff format controlnet.py --------- Co-authored-by: Lincoln Stein <lincoln.stein@gmail.com>
1 parent 10eebb6 commit 8d6e005

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

invokeai/backend/model_manager/configs/controlnet.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -88,7 +88,9 @@ def from_model_on_disk(cls, mod: ModelOnDisk, override_fields: dict[str, Any]) -
8888

8989
cls._validate_base(mod)
9090

91-
return cls(**override_fields)
91+
repo_variant = {"repo_variant": override_fields.get("repo_variant", cls._get_repo_variant_or_raise(mod))}
92+
args = override_fields | repo_variant
93+
return cls(**args)
9294

9395
@classmethod
9496
def _validate_base(cls, mod: ModelOnDisk) -> None:

0 commit comments

Comments
 (0)