anton-l HF staff commited on
Commit
96f0381
·
1 Parent(s): 70ed895

Update pipeline_glide.py

Browse files
Files changed (1) hide show
  1. 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.upscale_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)
 
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)