Update modules/translation/nllb_inference.py
Browse files
modules/translation/nllb_inference.py
CHANGED
@@ -50,7 +50,7 @@ class NLLBInference(TranslationBase):
|
|
50 |
|
51 |
if model_size != self.current_model_size or self.model is None:
|
52 |
print("\nInitializing NLLB Model..\n")
|
53 |
-
progress(0, desc="Initializing NLLB Model
|
54 |
self.current_model_size = model_size
|
55 |
local_files_only = self.is_model_exists(self.current_model_size)
|
56 |
self.model = AutoModelForSeq2SeqLM.from_pretrained(pretrained_model_name_or_path=model_size,
|
|
|
50 |
|
51 |
if model_size != self.current_model_size or self.model is None:
|
52 |
print("\nInitializing NLLB Model..\n")
|
53 |
+
progress(0, desc="Initializing NLLB Model...")
|
54 |
self.current_model_size = model_size
|
55 |
local_files_only = self.is_model_exists(self.current_model_size)
|
56 |
self.model = AutoModelForSeq2SeqLM.from_pretrained(pretrained_model_name_or_path=model_size,
|