Upload folder using huggingface_hub
Browse files- model_onnx.py +1 -1
model_onnx.py
CHANGED
@@ -77,7 +77,7 @@ class IndicASRModel(PreTrainedModel):
|
|
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)):
|
|
|
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.config.PRED_RNN_HIDDEN_DIM), dtype=np.float32))
|
81 |
|
82 |
# Iterate over time steps (T)
|
83 |
for t in range(joint_enc.size(1)):
|