Skip to content

Commit a379094

Browse files
authored
fix(itn): compose before star to speedup (#64)
1 parent a83fb69 commit a379094

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

itn/chinese/inverse_normalizer.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -66,12 +66,12 @@ def build_tagger(self):
6666
| add_weight(Time().tagger, 1.05)
6767
| add_weight(Cardinal(self.convert_number, self.enable_0_to_9).tagger, 1.06)
6868
| add_weight(Math().tagger, 1.10)
69-
| add_weight(Char().tagger, 100)).optimize().star
70-
# remove the last space
71-
tagger @= self.build_rule(delete(' '), '', '[EOS]')
69+
| add_weight(Char().tagger, 100)).optimize()
7270

7371
processor = PreProcessor(remove_interjections=True).processor
74-
self.tagger = processor @ tagger.optimize()
72+
tagger = (processor @ tagger).star
73+
# remove the last space
74+
self.tagger = tagger @ self.build_rule(delete(' '), '', '[EOS]')
7575

7676
def build_verbalizer(self):
7777
verbalizer = (Cardinal(self.convert_number, self.enable_0_to_9).verbalizer

0 commit comments

Comments
 (0)