Update app.py
Browse files
app.py
CHANGED
|
@@ -171,6 +171,7 @@ def infer(
|
|
| 171 |
generator=generator
|
| 172 |
).images[0]
|
| 173 |
else:
|
|
|
|
| 174 |
with torch.no_grad():
|
| 175 |
sd_image = pipe(
|
| 176 |
prompt=enhanced_prompt, # This conversion is fine
|
|
@@ -179,6 +180,7 @@ def infer(
|
|
| 179 |
num_inference_steps=num_inference_steps,
|
| 180 |
width=width,
|
| 181 |
height=height,
|
|
|
|
| 182 |
generator=generator
|
| 183 |
).images[0]
|
| 184 |
print('-- got image --')
|
|
|
|
| 171 |
generator=generator
|
| 172 |
).images[0]
|
| 173 |
else:
|
| 174 |
+
print('-- generating image --')
|
| 175 |
with torch.no_grad():
|
| 176 |
sd_image = pipe(
|
| 177 |
prompt=enhanced_prompt, # This conversion is fine
|
|
|
|
| 180 |
num_inference_steps=num_inference_steps,
|
| 181 |
width=width,
|
| 182 |
height=height,
|
| 183 |
+
latent=None,
|
| 184 |
generator=generator
|
| 185 |
).images[0]
|
| 186 |
print('-- got image --')
|