Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -106,6 +106,7 @@ def generate_image(
|
|
| 106 |
num_inference_steps=steps,
|
| 107 |
strength=strength,
|
| 108 |
generator = generator,
|
|
|
|
| 109 |
)
|
| 110 |
else:
|
| 111 |
image = pipe(
|
|
@@ -116,6 +117,7 @@ def generate_image(
|
|
| 116 |
guidance_scale=scale,
|
| 117 |
num_inference_steps=steps,
|
| 118 |
generator = generator,
|
|
|
|
| 119 |
)
|
| 120 |
return image.images[0]
|
| 121 |
|
|
|
|
| 106 |
num_inference_steps=steps,
|
| 107 |
strength=strength,
|
| 108 |
generator = generator,
|
| 109 |
+
prompt_embeds = None,
|
| 110 |
)
|
| 111 |
else:
|
| 112 |
image = pipe(
|
|
|
|
| 117 |
guidance_scale=scale,
|
| 118 |
num_inference_steps=steps,
|
| 119 |
generator = generator,
|
| 120 |
+
prompt_embeds = None,
|
| 121 |
)
|
| 122 |
return image.images[0]
|
| 123 |
|