Update Storydiffusion_node.py

fix bug
2024-05-30 21:23:36 +08:00
parent 8f7184da81
commit fe9a93f6ba
1 changed files with 1 additions and 1 deletions
@@ -1253,7 +1253,7 @@ class Storydiffusion_Img2Img:
        if device != "mps":
            pipe.enable_model_cpu_offload()
        unet = pipe.unet
-        cur_model_type = sd_type + "-" + "original"
+        cur_model_type = "Unstable" + "-" + "original"
        ### Insert PairedAttention
        for name in unet.attn_processors.keys():
            cross_attention_dim = (