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 |
|