Update whisper2.py

whisper2.py

@@ -10,8 +10,8 @@ device = "cuda:0" if torch.cuda.is_available() else "cpu"
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 print("[ INFO ] Device: ", device)
-torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
-
+#torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
+torch_dtype = torch.float32
 
 model = WhisperForConditionalGeneration.from_pretrained(MODEL_NAME, torch_dtype=torch_dtype).to(device)
 processor = WhisperProcessor.from_pretrained(MODEL_NAME)