Hugging Face's logo

Spaces:
VanguardAI
/
RealTime
Build error

App Files Community
VanguardAI commited on
Commit
ec99632
•
1 Parent(s): 61562a2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ torch_dtype = torch.float16
14
  # Whisper setup
15
  whisper_model_id = "openai/whisper-large-v3"
16
  whisper_model = AutoModelForSpeechSeq2Seq.from_pretrained(
17
- whisper_model_id, torch_dtype=torch_dtype, use_safetensors=True
18
  )
19
  whisper_processor = AutoProcessor.from_pretrained(whisper_model_id)
20
  whisper_pipe = pipeline(
 
14
  # Whisper setup
15
  whisper_model_id = "openai/whisper-large-v3"
16
  whisper_model = AutoModelForSpeechSeq2Seq.from_pretrained(
17
+ whisper_model_id, torch_dtype=torch_dtype, use_safetensors=True, low_cpu_mem_usage=True,
18
  )
19
  whisper_processor = AutoProcessor.from_pretrained(whisper_model_id)
20
  whisper_pipe = pipeline(