Spaces:

gokaygokay
/

Hunyuan3D-1.0

Running on Zero

gokaygokay commited on 6 days ago

Commit

f1bcd85

•

1 Parent(s): 2fb7bdc

diffusers

Files changed (1) hide show

module/ip_adapter/utils.py CHANGED Viewed

@@ -79,7 +79,7 @@ def load_adapter_to_pipe(
             if not isinstance(pretrained_model_path_or_dict, dict):
                 if pretrained_model_path_or_dict.endswith(".safetensors"):
                     state_dict = {"image_proj": {}, "ip_adapter": {}}
-                    with safe_open(pretrained_model_path_or_dict, framework="pt", device=unet.device) as f:
                         for key in f.keys():
                             if key.startswith("image_proj."):
                                 state_dict["image_proj"][key.replace("image_proj.", "")] = f.get_tensor(key)

             if not isinstance(pretrained_model_path_or_dict, dict):
                 if pretrained_model_path_or_dict.endswith(".safetensors"):
                     state_dict = {"image_proj": {}, "ip_adapter": {}}
+                    with safe_open(pretrained_model_path_or_dict, framework="pt", device=pipe.unet.device) as f:
                         for key in f.keys():
                             if key.startswith("image_proj."):
                                 state_dict["image_proj"][key.replace("image_proj.", "")] = f.get_tensor(key)