LAP-DEV commited on
Commit
bd30de2
·
verified ·
1 Parent(s): af8aea1

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,