Spaces:
Running
on
Zero
Running
on
Zero
Update adapter configuration in app.py to include 'lightning' alongside 'stage2' for enhanced inference capabilities.
Browse files
app.py
CHANGED
|
@@ -137,7 +137,7 @@ def infer(
|
|
| 137 |
print(f"Seed: {seed}, Steps: {num_inference_steps}, Guidance: {true_guidance_scale}, Size: {width}x{height}")
|
| 138 |
print("LoRA Weights - Stage2: 1.0")
|
| 139 |
|
| 140 |
-
pipe.set_adapters(["stage2"], adapter_weights=[1.0])
|
| 141 |
stage2_images = pipe(
|
| 142 |
image=[pil_image] if pil_image is not None else None,
|
| 143 |
prompt=STAGE2_PROMPT,
|
|
|
|
| 137 |
print(f"Seed: {seed}, Steps: {num_inference_steps}, Guidance: {true_guidance_scale}, Size: {width}x{height}")
|
| 138 |
print("LoRA Weights - Stage2: 1.0")
|
| 139 |
|
| 140 |
+
pipe.set_adapters(["lightning", "stage2"], adapter_weights=[1.0, 1.0])
|
| 141 |
stage2_images = pipe(
|
| 142 |
image=[pil_image] if pil_image is not None else None,
|
| 143 |
prompt=STAGE2_PROMPT,
|