Zeimoto commited on
Commit
76c37f0
1 Parent(s): fa6f424

cpu usage set as high

Browse files
Files changed (1) hide show
  1. speech2text.py +1 -1
speech2text.py CHANGED
@@ -14,7 +14,7 @@ def init_model_trans ():
14
  model_id = "openai/whisper-large-v3"
15
 
16
  model = AutoModelForSpeechSeq2Seq.from_pretrained(
17
- model_id, torch_dtype=torch_dtype, low_cpu_mem_usage=True, use_safetensors=True
18
  )
19
  model.to(device)
20
 
 
14
  model_id = "openai/whisper-large-v3"
15
 
16
  model = AutoModelForSpeechSeq2Seq.from_pretrained(
17
+ model_id, torch_dtype=torch_dtype, low_cpu_mem_usage=False, use_safetensors=True
18
  )
19
  model.to(device)
20