Spaces:

coralLight
/

Hyperparameters-are-all-you-need-UniPC-XL

Running on Zero

coralLight commited on 6 days ago

Commit

2522cf4

1 Parent(s): f1daac8

add lora support

Files changed (1) hide show

app.py CHANGED Viewed

@@ -183,7 +183,7 @@ def generate_image_with_steps(prompt
                                                             , use_afs=True)
     if not need_lora:
         pipe.set_adapters(["spo", "aqm"], adapter_weights=[0.0, 0.0])
-    elif num_inference_steps > 6:
         pipe.set_adapters(["spo", "aqm"], adapter_weights=[0.7, 0.7])
     else:
         pipe.set_adapters(["spo", "aqm"], adapter_weights=[0.25, 0.25])

                                                             , use_afs=True)
     if not need_lora:
         pipe.set_adapters(["spo", "aqm"], adapter_weights=[0.0, 0.0])
+    elif num_inference_steps > 6 and num_inference_steps <= 8:
         pipe.set_adapters(["spo", "aqm"], adapter_weights=[0.7, 0.7])
     else:
         pipe.set_adapters(["spo", "aqm"], adapter_weights=[0.25, 0.25])