tahirjm commited on
Commit
197a249
·
verified ·
1 Parent(s): f3e64ec

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. model_onnx.py +2 -2
model_onnx.py CHANGED
@@ -76,8 +76,8 @@ class IndicASRModel(PreTrainedModel):
76
  joint_enc = torch.from_numpy(joint_enc)
77
  # Initialize hypothesis with SOS token
78
  hyp = [self.config.SOS]
79
- prev_dec_state = (np.zeros((self.PRED_RNN_LAYERS, 1, self.PRED_RNN_HIDDEN_DIM), dtype=np.float32),
80
- np.zeros((self.PRED_RNN_LAYERS, 1, self.PRED_RNN_HIDDEN_DIM), dtype=np.float32))
81
 
82
  # Iterate over time steps (T)
83
  for t in range(joint_enc.size(1)):
 
76
  joint_enc = torch.from_numpy(joint_enc)
77
  # Initialize hypothesis with SOS token
78
  hyp = [self.config.SOS]
79
+ prev_dec_state = (np.zeros((self.config.PRED_RNN_LAYERS, 1, self.config.PRED_RNN_HIDDEN_DIM), dtype=np.float32),
80
+ np.zeros((self.config.PRED_RNN_LAYERS, 1, self.PRED_RNN_HIDDEN_DIM), dtype=np.float32))
81
 
82
  # Iterate over time steps (T)
83
  for t in range(joint_enc.size(1)):