Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -31,7 +31,7 @@ def infer (prompt, inf_steps, guidance_scale, seed, lora_weigth, progress=gr.Pro
|
|
| 31 |
num_inference_steps=inf_steps,
|
| 32 |
guidance_scale = float(guidance_scale),
|
| 33 |
generator=generator,
|
| 34 |
-
cross_attention_kwargs={"scale":
|
| 35 |
).images[0]
|
| 36 |
return image
|
| 37 |
|
|
|
|
| 31 |
num_inference_steps=inf_steps,
|
| 32 |
guidance_scale = float(guidance_scale),
|
| 33 |
generator=generator,
|
| 34 |
+
cross_attention_kwargs={"scale": lora_weight}
|
| 35 |
).images[0]
|
| 36 |
return image
|
| 37 |
|