Skip to content

GitLab

Unverified Commit 3067da12 authored by Dhruv Nair's avatar Dhruv Nair Committed by GitHub
Browse files

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

update
parent 6bceaea3
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/diffusers/loaders/single_file_utils.py
View file @ 3067da12
...@@ -876,7 +876,7 @@ def create_diffusers_controlnet_model_from_ldm( ...@@ -876,7 +876,7 @@ def create_diffusers_controlnet_model_from_ldm(
from ..models.modeling_utils import load_model_dict_into_meta from ..models.modeling_utils import load_model_dict_into_meta
unexpected_keys = 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: if controlnet._keys_to_ignore_on_load_unexpected is not None:
for pat in controlnet._keys_to_ignore_on_load_unexpected: for pat in controlnet._keys_to_ignore_on_load_unexpected:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment