Upload folder using huggingface_hub
Browse files- 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)):
|