VanguardAI commited on
Commit
61562a2
1 Parent(s): 0c6ed12

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
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, low_cpu_mem_usage=True, use_safetensors=True
18
  )
19
  whisper_processor = AutoProcessor.from_pretrained(whisper_model_id)
20
  whisper_pipe = pipeline(
@@ -27,7 +27,6 @@ whisper_pipe = pipeline(
27
  batch_size=16,
28
  return_timestamps=True,
29
  torch_dtype=torch_dtype,
30
- device=device,
31
  )
32
 
33
  # LLaMa3-8B setup
 
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(
 
27
  batch_size=16,
28
  return_timestamps=True,
29
  torch_dtype=torch_dtype,
 
30
  )
31
 
32
  # LLaMa3-8B setup