hf cuda issue

.gitignore

@@ -7,4 +7,7 @@ diffusion/__pycache__/
 DINO_EMA_11M_b50_lr1e-5_epoch6_step320k.ckpt
 sam_vit_h_4b8939.pth
 vae-ft-mse-840000-ema-pruned.ckpt
-how_to_videos/
+how_to_videos/
+debug.png
+debug*.png
+*.pem

app.py

@@ -724,7 +724,7 @@ def ready_sample(img_cropped, img_original, ex_mask, inpaint_mask, keypts, keypt
         img,
         keypts,
         hand_mask,
-        device=device,
+        device="cuda",
         target_size=(256, 256),
         latent_size=(32, 32),
     ):
@@ -760,7 +760,7 @@ def ready_sample(img_cropped, img_original, ex_mask, inpaint_mask, keypts, keypt
         img,
         keypts,
         hand_mask * (1 - inpaint_mask),
-        device=pre_device,
+        # device=pre_device,
         target_size=opts.image_size,
         latent_size=opts.latent_size,
     )