Merge pull request #114 from johnbenac/16_to_32_for_1070

Made compatible with older GPUs (1070)
This commit is contained in:
erew123 authored and GitHub committed 2024-03-10 15:12:42 +00:00
commit b8c0db6f56
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": []}