Lum4yx commited on
Commit
c96cde1
·
verified ·
1 Parent(s): ceb8eda

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 if torch.cuda.is_available() else torch.float32
14
  model_id = "openai/whisper-small"
15
 
16
  model = AutoModelForSpeechSeq2Seq.from_pretrained(
17
- model_id, torch_dtype=torch_dtype, low_cpu_mem_usage=True, use_safentensors=True
18
  )
19
  model.to(device)
20
  processor = AutoProcessor.from_pretrained(model_id)
 
14
  model_id = "openai/whisper-small"
15
 
16
  model = AutoModelForSpeechSeq2Seq.from_pretrained(
17
+ model_id, torch_dtype=torch_dtype, low_cpu_mem_usage=True
18
  )
19
  model.to(device)
20
  processor = AutoProcessor.from_pretrained(model_id)