hf cuda issue

app.py

@@ -765,7 +765,7 @@ def ready_sample(img_cropped, img_original, ex_mask, inpaint_mask, keypts, keypt
         target_size=opts.image_size,
         latent_size=opts.latent_size,
     )
-    latent = opts.latent_scaling_factor * autoencoder.encode(image.cuda()).sample()
+    latent = opts.latent_scaling_factor * autoencoder.encode(image.cuda()).sample().to(pre_device)
     target_cond = torch.cat([heatmaps, torch.zeros_like(mask)], 1)
     ref_cond = torch.cat([latent, heatmaps, mask], 1)
     ref_cond = torch.zeros_like(ref_cond)