Fix `load_model_dict_into_meta` for ControlNet `from_single_file` (#7034)

update

Fix `load_model_dict_into_meta` for ControlNet `from_single_file` (#7034)
update
3067da12 · Dhruv Nair · GitHub · 6bceaea3 · 3067da12
Unverified Commit 3067da12 authored Feb 20, 2024 by Dhruv Nair Committed by GitHub Feb 20, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/diffusers/loaders/single_file_utils.py src/diffusers/loaders/single_file_utils.py +1 -1

No files found.
--- a/src/diffusers/loaders/single_file_utils.py
+++ b/src/diffusers/loaders/single_file_utils.py
@@ -876,7 +876,7 @@ def create_diffusers_controlnet_model_from_ldm(
        from ..models.modeling_utils import load_model_dict_into_meta

        unexpected_keys = load_model_dict_into_meta(
-            controlnet, diffusers_format_controlnet_checkpoint, torch_dtype=torch_dtype
+            controlnet, diffusers_format_controlnet_checkpoint, dtype=torch_dtype
        )
        if controlnet._keys_to_ignore_on_load_unexpected is not None:
            for pat in controlnet._keys_to_ignore_on_load_unexpected: