Update kaggle_florence_gpu_1.py

kaggle_florence_gpu_1.py

@@ -23,9 +23,9 @@ from utils.florencegpu1 import load_florence_model, run_florence_inference, \
 from utils.modes import IMAGE_INFERENCE_MODES, IMAGE_OPEN_VOCABULARY_DETECTION_MODE, \
     IMAGE_CAPTION_GROUNDING_MASKS_MODE, VIDEO_INFERENCE_MODES
 from utils.sam import load_sam_image_model, run_sam_inference, load_sam_video_model
-DEVICE = torch.device("cuda")
-DEVICE = [torch.device(f'cuda:{i}') for i in range(torch.cuda.device_count())][-1]
-#DEVICE = [torch.device(f'cuda:{i}') for i in range(torch.cuda.device_count())][0]
+#DEVICE = torch.device("cuda")
+#DEVICE = [torch.device(f'cuda:{i}') for i in range(torch.cuda.device_count())][-1]
+DEVICE = [torch.device(f'cuda:{i}') for i in range(torch.cuda.device_count())][0]
 
 torch.autocast(device_type="cuda", dtype=torch.bfloat16).__enter__()
 if torch.cuda.get_device_properties(0).major >= 8: