Spaces:
Runtime error
Runtime error
Update inversion_utils.py
Browse files- 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 |
|