cuio commited on
Commit
e908c26
·
verified ·
1 Parent(s): cadd646

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -182,7 +182,7 @@ def generate(prompt: str, seed: int, randomize: bool):
182
  )
183
 
184
  with torch.no_grad(), torch.autocast("cpu", dtype=torch.bfloat16):
185
- lat = _state["model"].sample_images_with_grad(noise, ctx, {"num_steps": 1})
186
  pix = _state["vae"].detokenize(lat)
187
 
188
  arr = (pix[0].float().clamp(0, 1) * 255).round().byte()
 
182
  )
183
 
184
  with torch.no_grad(), torch.autocast("cpu", dtype=torch.bfloat16):
185
+ lat = _state["model"].sample_images_with_grad(noise, ctx, {"num_steps": 2})
186
  pix = _state["vae"].detokenize(lat)
187
 
188
  arr = (pix[0].float().clamp(0, 1) * 255).round().byte()