Lifeinhockey commited on
Commit
8feab8c
·
verified ·
1 Parent(s): 32728c8

Update app.py

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