update

app.py

@@ -122,6 +122,7 @@ def sample_euler_ancestral(model, x_0, x_1, sigmas, height, width, extra_args=No
 
     mask_list = []
     image_list = []
+    model.to(device)
     for i in trange(len(sigmas) - 1, disable=disable):
         denoised_0, denoised_1 = model(x_0, x_1, sigmas[i] * s_in, **extra_args)
         image_list.append(denoised_0)