Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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":
|
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!")
|