Fix a bug in the state dict judgment in ip_adapter.py. (#10095)

* fix a judging state dict bug in ip_adapter.py * make --------- Co-authored-by: hlky <[email protected]>
huggingface · Dec 5, 2024 · 73dac0c · 73dac0c
1 parent 04bba38
commit 73dac0c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/diffusers/loaders/ip_adapter.py b/src/diffusers/loaders/ip_adapter.py
@@ -187,7 +187,7 @@ def load_ip_adapter(
                 state_dict = pretrained_model_name_or_path_or_dict
 
             keys = list(state_dict.keys())
-            if keys != ["image_proj", "ip_adapter"]:
+            if "image_proj" not in keys and "ip_adapter" not in keys:
                 raise ValueError("Required keys are (`image_proj` and `ip_adapter`) missing from the state dict.")
 
             state_dicts.append(state_dict)