amildravid4292 commited on
Commit
77ce3b8
·
verified ·
1 Parent(s): c3bf238

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -227,7 +227,7 @@ class main():
227
  #guidance
228
  noise_pred_uncond, noise_pred_text = noise_pred.chunk(2)
229
  noise_pred = noise_pred_uncond + guidance_scale * (noise_pred_text - noise_pred_uncond)
230
- latents = noise_scheduler.step(noise_pred, t, latents).prev_sample
231
 
232
  latents = 1 / 0.18215 * latents
233
  image = self.vae.decode(latents).sample
 
227
  #guidance
228
  noise_pred_uncond, noise_pred_text = noise_pred.chunk(2)
229
  noise_pred = noise_pred_uncond + guidance_scale * (noise_pred_text - noise_pred_uncond)
230
+ latents = self.noise_scheduler.step(noise_pred, t, latents).prev_sample
231
 
232
  latents = 1 / 0.18215 * latents
233
  image = self.vae.decode(latents).sample