amildravid4292 commited on
Commit
cf8bccb
·
verified ·
1 Parent(s): 6b4a672

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -120,7 +120,7 @@ def inference( prompt, negative_prompt, guidance_scale, ddim_steps, seed):
120
  #guidance
121
  noise_pred_uncond, noise_pred_text = noise_pred.chunk(2)
122
  noise_pred = noise_pred_uncond + guidance_scale * (noise_pred_text - noise_pred_uncond)
123
- latents = noise_scheduler.step(noise_pred, t, latents).prev_sample
124
 
125
  latents = 1 / 0.18215 * latents
126
  image = vae.value.decode(latents).sample
 
120
  #guidance
121
  noise_pred_uncond, noise_pred_text = noise_pred.chunk(2)
122
  noise_pred = noise_pred_uncond + guidance_scale * (noise_pred_text - noise_pred_uncond)
123
+ latents = noise_scheduler.value.step(noise_pred, t, latents).prev_sample
124
 
125
  latents = 1 / 0.18215 * latents
126
  image = vae.value.decode(latents).sample