Spaces:

linoyts
/

edit_friendly_ddpm_inversion

Running on A10G

linoyts HF Staff commited on Nov 24, 2024

Commit

437dac5

verified ·

1 Parent(s): fcfe1ba

Update inversion_utils.py

Files changed (1) hide show

inversion_utils.py CHANGED Viewed

@@ -143,7 +143,7 @@ def inversion_forward_process(model, x0,
     timesteps = model.scheduler.timesteps.to(model.device)
     variance_noise_shape = (
         num_inference_steps,
-        model.unet.in_channels,
         model.unet.sample_size,
         model.unet.sample_size)
     if etas is None or (type(etas) in [int, float] and etas == 0):

     timesteps = model.scheduler.timesteps.to(model.device)
     variance_noise_shape = (
         num_inference_steps,
+        model.unet.config.in_channels,
         model.unet.sample_size,
         model.unet.sample_size)
     if etas is None or (type(etas) in [int, float] and etas == 0):