fix dict order (#5109)

facebookresearch · May 19, 2023 · bfd9dc6 · bfd9dc6
1 parent 5ecbbf5
commit bfd9dc6
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/examples/speech_recognition/new/decoders/flashlight_decoder.py b/examples/speech_recognition/new/decoders/flashlight_decoder.py
@@ -40,6 +40,7 @@
         Trie,
     )
     from flashlight.lib.text.dictionary import create_word_dict, load_words
+    from flashlight.lib.text.dictionary import Dictionary as flDictionary
 except ImportError:
     warnings.warn(
         "flashlight python bindings are required to use this functionality. "
@@ -102,8 +103,9 @@ def __init__(self, cfg: FlashlightDecoderConfig, tgt_dict: Dictionary) -> None:
         else:
             assert self.unitlm, "Lexicon-free decoding requires unit LM"
 
-            d = {w: [[w]] for w in tgt_dict.symbols}
-            self.word_dict = create_word_dict(d)
+            self.word_dict = flDictionary()
+            for sym in tgt_dict.symbols:
+                self.word_dict.add_entry(sym, tgt_dict.index(sym))
             self.lm = KenLM(cfg.lmpath, self.word_dict)
             self.decoder_opts = LexiconFreeDecoderOptions(
                 beam_size=cfg.beam,