yeq6x commited on
Commit
3d05df7
·
1 Parent(s): b44c75f

Update adapter configuration in app.py to include 'lightning' alongside 'stage2' for enhanced inference capabilities.

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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,