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 |
|