From 1f7842ef5a41d660c49982b1b8f6abd44fc2ab6f Mon Sep 17 00:00:00 2001 From: johnbenac Date: Sat, 9 Mar 2024 20:42:01 -0500 Subject: [PATCH] Made compatible with older GPUs (1070) --- finetune.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/finetune.py b/finetune.py index f35fc43..7f9026b 100644 --- a/finetune.py +++ b/finetune.py @@ -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": []}