Update pipeline_glide.py
Browse files- pipeline_glide.py +1 -1
pipeline_glide.py
CHANGED
|
@@ -792,7 +792,7 @@ class GLIDE(DiffusionPipeline):
|
|
| 792 |
frac = (model_var_values + 1) / 2
|
| 793 |
model_log_variance = frac * max_log + (1 - frac) * min_log
|
| 794 |
|
| 795 |
-
pred_prev_image = self.
|
| 796 |
noise_residual, image, t, num_inference_steps_upscale, eta, use_clipped_residual=True
|
| 797 |
)
|
| 798 |
noise = torch.randn(image.shape, generator=generator).to(torch_device)
|
|
|
|
| 792 |
frac = (model_var_values + 1) / 2
|
| 793 |
model_log_variance = frac * max_log + (1 - frac) * min_log
|
| 794 |
|
| 795 |
+
pred_prev_image = self.text_noise_scheduler.step(
|
| 796 |
noise_residual, image, t, num_inference_steps_upscale, eta, use_clipped_residual=True
|
| 797 |
)
|
| 798 |
noise = torch.randn(image.shape, generator=generator).to(torch_device)
|