forked from deepwavedigital/radio-asr
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathnemo-1.3.patch
25 lines (24 loc) · 898 Bytes
/
nemo-1.3.patch
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
diff --git a/nemo/collections/asr/modules/beam_search_decoder.py b/nemo/collections/asr/modules/beam_search_decoder.py
index b39804ae..9c3464b7 100644
--- a/nemo/collections/asr/modules/beam_search_decoder.py
+++ b/nemo/collections/asr/modules/beam_search_decoder.py
@@ -82,7 +82,6 @@ class BeamSearchDecoderWithLM(NeuralModule):
self.cutoff_top_n = cutoff_top_n
self.input_tensor = input_tensor
- @typecheck(ignore_collections=True)
@torch.no_grad()
def forward(self, log_probs, log_probs_length):
probs_list = log_probs
diff --git a/requirements/requirements_nlp.txt b/requirements/requirements_nlp.txt
index 31cde32a..06e7f8e0 100644
--- a/requirements/requirements_nlp.txt
+++ b/requirements/requirements_nlp.txt
@@ -9,7 +9,7 @@ rapidfuzz
gdown
megatron-lm==2.2.0
inflect
-sacrebleu[ja]
+sacrebleu
sacremoses>=0.0.43
nltk==3.6.2
wordninja==2.0.0