Made compatible with older GPUs (1070)

This commit is contained in:
johnbenac authored and GitHub committed 2024-03-09 20:42:01 -05:00
1 parent 9bb0d1fa58
commit 1f7842ef5a
1 file changed
+1 -1
+1 -1
View File
@@ -110,7 +110,7 @@ def format_audio_list(target_language, whisper_model, out_path, eval_split_numbe
device = "cuda" if torch.cuda.is_available() else "cpu"
print("[FINETUNE] Loading Whisper Model:", whisper_model)
asr_model = WhisperModel(whisper_model, device=device, compute_type="float16")
asr_model = WhisperModel(whisper_model, device=device, compute_type="float32")
metadata = {"audio_file": [], "text": [], "speaker_name": []}