hf cuda issue

app.py

@@ -395,7 +395,7 @@ def get_ref_anno(img, keypts, use_mask, use_pose):
     print(f"autoencoder encoder before operating max: {min([p.min() for p in autoencoder.encoder.parameters()])}")
     print(f"autoencoder encoder before operating min: {max([p.max() for p in autoencoder.encoder.parameters()])}")
     print(f"autoencoder encoder before operating dtype: {next(autoencoder.encoder.parameters()).dtype}")
-    latent = opts.latent_scaling_factor * autoencoder.encode(image).sample()
+    latent = opts.latent_scaling_factor * autoencoder.encode(image.cuda()).sample().to(pre_device)
     print(f"latent.max(): {latent.max()}, latent.min(): {latent.min()}")
     if no_mask:
         mask = torch.zeros_like(mask)