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

Update pipeline.py

Browse Source 
fix
This commit is contained in:
smthemex
2024-06-24 08:26:54 +08:00
committed by GitHub
parent 5f854270b5
commit 9ce13f32b6
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
streamv2v/pipeline.py
+2 -2
View File
@@ -106,8 +106,8 @@ class StreamV2V:
model_path = os.path.join(path_dir,"models")
model_lists = os.listdir(model_path)
if "lcm-lora-sdv1-5.safetensors" in model_lists: # 加入本地模型支持
pretrained_model_name_or_path_or_dict=get_instance_path(os.path.join(model_path,"lcm-lora-sdv1-5.safetensors"))
if "pytorch_lora_weights.safetensors" in model_lists: # 加入本地模型支持
pretrained_model_name_or_path_or_dict=get_instance_path(os.path.join(model_path,"pytorch_lora_weights.safetensors")) #lcm-lora-sdv1-5.safetensors
else:
snapshot_download("latent-consistency/lcm-lora-sdv1-5",local_dir=get_instance_path(model_path))
pretrained_model_name_or_path_or_dict=get_instance_path(os.path.join(model_path,"pytorch_lora_weights.safetensors"))