Yehor commited on
Commit
a8b1bf0
·
verified ·
1 Parent(s): 2c002d0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -92,7 +92,7 @@ for path in paths:
92
 
93
  # Transcribe the audio
94
  inputs = processor(audio_inputs, sampling_rate=sampling_rate).input_features
95
- features = torch.tensor(inputs).to(device)
96
 
97
  with torch.inference_mode():
98
  logits = asr_model(features).logits
 
92
 
93
  # Transcribe the audio
94
  inputs = processor(audio_inputs, sampling_rate=sampling_rate).input_features
95
+ features = torch.tensor(inputs).half().to(device)
96
 
97
  with torch.inference_mode():
98
  logits = asr_model(features).logits