shenjingwen commited on
Commit
434a82b
Β·
verified Β·
1 Parent(s): de90052

Update whisper_asr.py

Browse files
Files changed (1) hide show
  1. whisper_asr.py +5 -2
whisper_asr.py CHANGED
@@ -7,8 +7,11 @@ import os
7
  import time
8
 
9
  class WhisperAutomaticSpeechRecognizer:
10
- device = "cuda"
11
- compute_type = "int8" # change to if more gpu memory available
 
 
 
12
  batch_size = 4
13
  model = whisperx.load_model(
14
  "medium", device, language="en", compute_type=compute_type
 
7
  import time
8
 
9
  class WhisperAutomaticSpeechRecognizer:
10
+ # device = "cuda"
11
+ # compute_type = "int8" # change to if more gpu memory available
12
+ device = "cpu"
13
+ compute_type = "float32"
14
+
15
  batch_size = 4
16
  model = whisperx.load_model(
17
  "medium", device, language="en", compute_type=compute_type