Hugging Face's logo

fxmarty
/
whisper-static-shapes-onnx

Automatic Speech Recognition
Transformers
ONNX
whisper
Inference Endpoints
Model card Files Community
Félix Marty commited on
Commit
796499f
·
1 Parent(s): 46b9c37

update

Browse files
Files changed (1) hide show
  1. run_whisper_ort.py +2 -2
run_whisper_ort.py CHANGED
@@ -55,7 +55,7 @@ position_ids = np.array([[0]]).astype(np.int64)
55
  token = 50257 # see whisper config decoder_start_token_id
56
 
57
  decoder_inputs = {
58
- "encoder_hidden_states": encoder_outputs[0],
59
  }
60
 
61
  for i in range(num_hidden_layers):
@@ -69,7 +69,7 @@ for i in range(num_hidden_layers):
69
  while True:
70
  decoder_inputs["position_ids"] = position_ids
71
  decoder_inputs["decoder_attention_mask"] = decoder_attention_mask
72
- decoder_inputs["input_ids"] = np.array([[token]]).astype(np.int64)
73
 
74
  decoder_outputs = decoder.run(None, decoder_inputs)
75
 
 
55
  token = 50257 # see whisper config decoder_start_token_id
56
 
57
  decoder_inputs = {
58
+ "encoder_outputs": encoder_outputs[0],
59
  }
60
 
61
  for i in range(num_hidden_layers):
 
69
  while True:
70
  decoder_inputs["position_ids"] = position_ids
71
  decoder_inputs["decoder_attention_mask"] = decoder_attention_mask
72
+ decoder_inputs["decoder_input_ids"] = np.array([[token]]).astype(np.int64)
73
 
74
  decoder_outputs = decoder.run(None, decoder_inputs)
75