Update sam2_mask.py

sam2_mask.py

@@ -10,8 +10,8 @@ from sam2.build_sam import build_sam2
 from sam2.sam2_image_predictor import SAM2ImagePredictor
 
 
-# Remove all CUDA-specific configurations
-torch.autocast(device_type="cpu", dtype=torch.float32).__enter__()
+# # Remove all CUDA-specific configurations
+# torch.autocast(device_type="cpu", dtype=torch.float32).__enter__()
 
 def preprocess_image(image):
     return image, gr.State([]), gr.State([]), image