Spaces:

dezzman
/

diffusion_models

Running

dezzman commited on Feb 15

Commit

5d636be

verified ·

1 Parent(s): e8401d5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -158,7 +158,8 @@ def infer(
             'num_inference_steps': num_inference_steps,
             'width': width,
             'height': height,
-            'generator': generator
         }
         if cn_enable:

             'num_inference_steps': num_inference_steps,
             'width': width,
             'height': height,
+            'generator': generator,
+            'cross_attention_kwargs': {"scale": lora_scale},
         }
         if cn_enable: