From d1bc49d9734b88ebc8d9d51d7346b9ef995d3c8a Mon Sep 17 00:00:00 2001
From: smthemex <138738845+smthemex@users.noreply.github.com>
Date: Wed, 14 Aug 2024 09:27:45 +0800
Subject: [PATCH] Update pipeline_v2.py

---
 utils/pipeline_v2.py | 46 ++++++++++++++++++++++++++++++--------------
 1 file changed, 32 insertions(+), 14 deletions(-)

diff --git a/utils/pipeline_v2.py b/utils/pipeline_v2.py
index 7ecb4e9..f4fca07 100644
--- a/utils/pipeline_v2.py
+++ b/utils/pipeline_v2.py
@@ -202,19 +202,37 @@ class PhotoMakerStableDiffusionXLPipeline(StableDiffusionXLPipeline):
         }
 
         if not isinstance(pretrained_model_name_or_path_or_dict, dict):
-            model_file = _get_model_file(
-                pretrained_model_name_or_path_or_dict,
-                weights_name=weight_name,
-                cache_dir=cache_dir,
-                force_download=force_download,
-                resume_download=resume_download,
-                proxies=proxies,
-                local_files_only=local_files_only,
-                token=token,
-                revision=revision,
-                subfolder=subfolder,
-                user_agent=user_agent,
-            )
+            try:
+                model_file = _get_model_file(
+                    pretrained_model_name_or_path_or_dict,
+                    weights_name=weight_name,
+                    cache_dir=cache_dir,
+                    force_download=force_download,
+                    proxies=proxies,
+                    local_files_only=local_files_only,
+                    token=token,
+                    revision=revision,
+                    subfolder=subfolder,
+                    user_agent=user_agent,
+                )
+            except:
+                try:
+                    model_file = _get_model_file(
+                        pretrained_model_name_or_path_or_dict,
+                        weights_name=weight_name,
+                        cache_dir=cache_dir,
+                        force_download=force_download,
+                        resume_download=resume_download,
+                        proxies=proxies,
+                        local_files_only=local_files_only,
+                        token=token,
+                        revision=revision,
+                        subfolder=subfolder,
+                        user_agent=user_agent,
+                    )
+                except:
+                    pass
+           
             if weight_name.endswith(".safetensors"):
                 state_dict = {"id_encoder": {}, "lora_weights": {}}
                 with safe_open(model_file, framework="pt", device="cpu") as f:
@@ -1067,4 +1085,4 @@ class PhotoMakerStableDiffusionXLPipeline(StableDiffusionXLPipeline):
         if not return_dict:
             return (image,)
         
-        return StableDiffusionXLPipelineOutput(images=image)
\ No newline at end of file
+        return StableDiffusionXLPipelineOutput(images=image)