Spaces:

fffiloni
/

sd-xl-custom-model

Sleeping

fffiloni commited on Jun 11, 2024

Commit

c46ad1b

verified ·

1 Parent(s): 2409154

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -96,7 +96,7 @@ def infer (custom_model, weight_name, prompt, inf_steps, guidance_scale, width,
             use_auth_token = True
         )
-    #pipe.fuse_lora(lora_weight)
     if seed < 0 :
         seed = random.randint(0, 423538377342)
@@ -113,7 +113,7 @@ def infer (custom_model, weight_name, prompt, inf_steps, guidance_scale, width,
         cross_attention_kwargs={"scale": lora_weight}
     ).images[0]
-    #pipe.unfuse_lora()
     return image, seed

             use_auth_token = True
         )
+    pipe.fuse_lora()
     if seed < 0 :
         seed = random.randint(0, 423538377342)
         cross_attention_kwargs={"scale": lora_weight}
     ).images[0]
+    pipe.unfuse_lora()
     return image, seed