ford442 commited on
Commit
6f8cde1
·
verified ·
1 Parent(s): 2d10cf0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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 --')