Spaces:

dbaranchuk
/

Scale-wise-Distillation

Running on Zero

dbaranchuk commited on 12 days ago

Commit

99b7ce8

verified ·

1 Parent(s): 08ba86b

Update inference.py

Files changed (1) hide show

inference.py CHANGED Viewed

@@ -230,7 +230,7 @@ def run(
             sigma_next = sigmas[i + 1]
             x0_pred = (latents - sigma * noise_pred)
             try:
-                x0_pred = torch.nn.functional.interpolate(x0_pred, size=scales[i + 1])
             except IndexError:
                 x0_pred = x0_pred
             noise = torch.randn(x0_pred.shape, generator=generator).to('cuda').half()

             sigma_next = sigmas[i + 1]
             x0_pred = (latents - sigma * noise_pred)
             try:
+                x0_pred = torch.nn.functional.interpolate(x0_pred, size=scales[i + 1], mode='bicubic')
             except IndexError:
                 x0_pred = x0_pred
             noise = torch.randn(x0_pred.shape, generator=generator).to('cuda').half()