Michael Natanael commited on
Commit
816f4f3
·
1 Parent(s): 7a5af5b

change model to turbo

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -51,7 +51,8 @@ model.eval()
51
 
52
  # === INITIAL SETUP: Faster Whisper ===
53
  # https://github.com/SYSTRAN/faster-whisper
54
- faster_whisper_model_size = "large-v3"
 
55
 
56
  # Run on GPU with FP16
57
  # model = WhisperModel(model_size, device="cuda", compute_type="float16")
@@ -61,8 +62,7 @@ faster_whisper_model_size = "large-v3"
61
  faster_whisper_model = WhisperModel(
62
  faster_whisper_model_size,
63
  device="cpu",
64
- compute_type="int8",
65
- cpu_threads=4
66
  )
67
 
68
 
 
51
 
52
  # === INITIAL SETUP: Faster Whisper ===
53
  # https://github.com/SYSTRAN/faster-whisper
54
+ # faster_whisper_model_size = "large-v3"
55
+ faster_whisper_model_size = "turbo"
56
 
57
  # Run on GPU with FP16
58
  # model = WhisperModel(model_size, device="cuda", compute_type="float16")
 
62
  faster_whisper_model = WhisperModel(
63
  faster_whisper_model_size,
64
  device="cpu",
65
+ compute_type="int8"
 
66
  )
67
 
68