Update Storydiffusion_node.py

This commit is contained in:
smthemex
2024-08-23 14:03:28 +08:00
committed by GitHub
parent 6d0463c5c0
commit 982740dc2c
+1 -1
View File
@@ -1701,7 +1701,7 @@ class Storydiffusion_Model_Loader:
pipe.text_encoder_2 = text_encoder_2
pipe.enable_model_cpu_offload()
if lora != "none":
if lora:
pipe.load_lora_weights(lora_path)
pipe.fuse_lora(lora_scale=0.125)
pipe.unload_lora_weights()