jiuface commited on
Commit
43dc2ed
·
verified ·
1 Parent(s): df50fe7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -92,7 +92,7 @@ def generate_image(orginal_image, prompt, adapter_names, steps, seed, image_str
92
  width=width,
93
  height=height,
94
  generator=generator,
95
- joint_attention_kwargs={"scale": lora_scale}
96
  ).images[0]
97
  else:
98
  generated_image = pipe(
@@ -103,6 +103,7 @@ def generate_image(orginal_image, prompt, adapter_names, steps, seed, image_str
103
  height=height,
104
  max_sequence_length=512,
105
  generator=generator,
 
106
  ).images[0]
107
 
108
  progress(99, "Generate image success!")
 
92
  width=width,
93
  height=height,
94
  generator=generator,
95
+ joint_attention_kwargs={"scale": 1}
96
  ).images[0]
97
  else:
98
  generated_image = pipe(
 
103
  height=height,
104
  max_sequence_length=512,
105
  generator=generator,
106
+ joint_attention_kwargs={"scale": 1}
107
  ).images[0]
108
 
109
  progress(99, "Generate image success!")