gpt-omni commited on
Commit
69a5822
1 Parent(s): 58227c7
Files changed (1) hide show
  1. inference.py +0 -1
inference.py CHANGED
@@ -138,7 +138,6 @@ def get_input_ids_whisper_ATBatch(mel, leng, whispermodel, device):
138
  return torch.stack([audio_feature, audio_feature]), stacked_inputids
139
 
140
 
141
- @spaces.GPU
142
  def load_audio(path):
143
  audio = whisper.load_audio(path)
144
  duration_ms = (len(audio) / 16000) * 1000
 
138
  return torch.stack([audio_feature, audio_feature]), stacked_inputids
139
 
140
 
 
141
  def load_audio(path):
142
  audio = whisper.load_audio(path)
143
  duration_ms = (len(audio) / 16000) * 1000