Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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
|