Spaces:
Runtime error
Runtime error
Zeimoto
commited on
Commit
•
76c37f0
1
Parent(s):
fa6f424
cpu usage set as high
Browse files- 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=
|
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 |
|