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