Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -158,9 +158,9 @@ def infer(
|
|
| 158 |
'num_inference_steps': num_inference_steps,
|
| 159 |
'width': width,
|
| 160 |
'height': height,
|
| 161 |
-
'generator': generator,
|
| 162 |
-
|
| 163 |
-
|
| 164 |
|
| 165 |
if cn_enable:
|
| 166 |
params['image'] = process_control_image(
|
|
|
|
| 158 |
'num_inference_steps': num_inference_steps,
|
| 159 |
'width': width,
|
| 160 |
'height': height,
|
| 161 |
+
'generator': generator,
|
| 162 |
+
'cross_attention_kwargs': {"scale": lora_scale},
|
| 163 |
+
}
|
| 164 |
|
| 165 |
if cn_enable:
|
| 166 |
params['image'] = process_control_image(
|