Update modules/translation/nllb_inference.py

modules/translation/nllb_inference.py

@@ -16,7 +16,7 @@ class NLLBInference(TranslationBase):
             output_dir=output_dir
         )
         self.tokenizer = None
-        self.available_models = ["facebook/nllb-200-3.3B", "facebook/nllb-200-1.3B", "facebook/nllb-200-distilled-600M"]
+        self.available_models = ["facebook/nllb-200-3.3B", "facebook/nllb-200-1.3B", "facebook/nllb-200-distilled-1.3B", "facebook/nllb-200-distilled-600M"]
         self.available_source_langs = list(NLLB_AVAILABLE_LANGS.keys())
         self.available_target_langs = list(NLLB_AVAILABLE_LANGS.keys())
         self.pipeline = None