anton-l HF staff commited on
Commit
0f49cfb
·
1 Parent(s): bbd7574

remove image postproc

Browse files
Files changed (1) hide show
  1. modeling_glide.py +1 -1
modeling_glide.py CHANGED
@@ -223,6 +223,6 @@ class GLIDE(DiffusionPipeline):
223
  sampled_prev_image = prev_image + prev_variance
224
  image = sampled_prev_image
225
 
226
- image = image[0].permute(1, 2, 0)
227
 
228
  return image
 
223
  sampled_prev_image = prev_image + prev_variance
224
  image = sampled_prev_image
225
 
226
+ image = image.permute(0, 2, 3, 1)
227
 
228
  return image