Merge pull request #203 from ebraraktas/fix/conformer-tflite-conversion

Fix/conformer tflite conversion

Author: nglehuy

examples/conformer/tflite.py

@@ -34,7 +34,10 @@
 
 parser.add_argument("--saved", type=str, default=None, help="Path to saved model")
 
-parser.add_argument("--subwords", type=str, default=None, help="Use subwords")
+parser.add_argument("--subwords", action="store_true", help="Use subwords")
+
+parser.add_argument("--vocabulary", type=str, default=None, required=False,
+                    help="Path to vocabulary. Overrides path in config, if given.")
 
 parser.add_argument("output", type=str, default=None, help="TFLite file path to be exported")
 
@@ -45,6 +48,9 @@
 config = Config(args.config)
 speech_featurizer = TFSpeechFeaturizer(config.speech_config)
 
+if args.vocabulary is not None:
+    config.decoder_config["vocabulary"] = args.vocabulary
+
 if args.subwords:
     text_featurizer = SubwordFeaturizer(config.decoder_config)
 else: