Spaces:
Running
on
A10G
Running
on
A10G
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 |
|