linoyts HF staff commited on
Commit
92455b3
1 Parent(s): f3f25c0

Update inversion_utils.py

Browse files
Files changed (1) hide show
  1. inversion_utils.py +1 -1
inversion_utils.py CHANGED
@@ -73,7 +73,7 @@ def sample_xts_from_x0(model, x0, num_inference_steps=50):
73
  betas = 1 - alphas
74
  variance_noise_shape = (
75
  num_inference_steps,
76
- model.unet.in_channels,
77
  model.unet.sample_size,
78
  model.unet.sample_size)
79
 
 
73
  betas = 1 - alphas
74
  variance_noise_shape = (
75
  num_inference_steps,
76
+ model.unet.config.in_channels,
77
  model.unet.sample_size,
78
  model.unet.sample_size)
79