Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -529,6 +529,7 @@ def infer(
|
|
529 |
'width': width,
|
530 |
'height': height,
|
531 |
'generator': generator,
|
|
|
532 |
}
|
533 |
|
534 |
image = pipe_LCM(**params).images[0]
|
|
|
529 |
'width': width,
|
530 |
'height': height,
|
531 |
'generator': generator,
|
532 |
+
'cross_attention_kwargs': {"scale": lora_scale},
|
533 |
}
|
534 |
|
535 |
image = pipe_LCM(**params).images[0]
|