Argument and dtype fix

pipeline.py

@@ -244,11 +244,9 @@ class SwDPipeline(StableDiffusion3Pipeline):
                 sigma = sigmas[i]
                 sigma_next = sigmas[i + 1]
                 x0_pred = (latents - sigma * noise_pred)
-                try:
+                if scales and i + 1 < len(scales):
                     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()
+                noise = torch.randn(x0_pred.shape, generator=generator, device=device, dtype=x0_pred.dtype)
                 latents = (1 - sigma_next) * x0_pred + sigma_next * noise
 
                 if latents.dtype != latents_dtype: