killwithabass commited on
Commit
a09a841
·
verified ·
1 Parent(s): 77181d8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -52,13 +52,13 @@ def run_lora(prompt, negative_prompt, cfg_scale, steps, randomize_seed, seed, wi
52
  # Generate image using the pipeline
53
  image = pipe(
54
  prompt=f"{prompt} {trigger_word}",
55
- negative_prompt,
56
  num_inference_steps=steps,
57
  guidance_scale=cfg_scale,
58
  width=width,
59
  height=height,
60
  generator=generator,
61
- joint_attention_kwargs={"scale": lora_scale}
62
  ).images[0]
63
 
64
  # Save the image to a file with a unique name in /tmp directory
 
52
  # Generate image using the pipeline
53
  image = pipe(
54
  prompt=f"{prompt} {trigger_word}",
55
+ negative_prompt=negative_prompt,
56
  num_inference_steps=steps,
57
  guidance_scale=cfg_scale,
58
  width=width,
59
  height=height,
60
  generator=generator,
61
+ joint_attention_kwargs={"scale": lora_scale},
62
  ).images[0]
63
 
64
  # Save the image to a file with a unique name in /tmp directory