Update pipeline_glide.py
Browse files- pipeline_glide.py +1 -3
pipeline_glide.py
CHANGED
|
@@ -792,9 +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.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)
|
| 799 |
variance = torch.exp(0.5 * model_log_variance) * noise
|
| 800 |
|
|
|
|
| 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(noise_residual, image, t)
|
|
|
|
|
|
|
| 796 |
noise = torch.randn(image.shape, generator=generator).to(torch_device)
|
| 797 |
variance = torch.exp(0.5 * model_log_variance) * noise
|
| 798 |
|