nodes/smthemex-ComfyUI_StoryDiffusion
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update pipeline_v2.py

Browse Source
This commit is contained in:
smthemex
2024-08-14 09:27:45 +08:00
committed by GitHub
parent 155252be5c
commit d1bc49d973
1 changed files with 32 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
utils/pipeline_v2.py
+32 -14
View File
@@ -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)
return StableDiffusionXLPipelineOutput(images=image)