Spaces:
Running
on
Zero
Running
on
Zero
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 --')
|