fix inpainting bug

app.py

@@ -741,10 +741,10 @@ def sample_inpaint(
     z = torch.randn(
         (N, opts.latent_dim, opts.latent_size[0], opts.latent_size[1]), device=device
     )
-    target_cond_N = target_cond.repeat(N, 1, 1, 1)
-    ref_cond_N = ref_cond.repeat(N, 1, 1, 1)
+    target_cond_N = target_cond.repeat(N, 1, 1, 1).to(z.device)
+    ref_cond_N = ref_cond.repeat(N, 1, 1, 1).to(z.device)
     # novel view synthesis mode = off
-    nvs = torch.zeros(N, dtype=torch.int, device=z.device)
+    nvs = torch.zeros(N, dtype=torch.int, device=device)
     z = torch.cat([z, z], 0)
     model_kwargs = dict(
         target_cond=torch.cat([target_cond_N, torch.zeros_like(target_cond_N)]),
@@ -756,8 +756,8 @@ def sample_inpaint(
     samples, _ = diffusion.inpaint_p_sample_loop(
         model.forward_with_cfg,
         z.shape,
-        latent,
-        inpaint_latent_mask,
+        latent.to(z.device),
+        inpaint_latent_mask.to(z.device),
         z,
         clip_denoised=False,
         model_kwargs=model_kwargs,