Iker commited on
Commit
122365e
·
1 Parent(s): a4d80df

Print model name

Browse files
Files changed (1) hide show
  1. translate.py +2 -2
translate.py CHANGED
@@ -68,11 +68,11 @@ def main(
68
  mixed_precision=precision if precision != "32" else "no", split_batches=True
69
  )
70
 
71
- print("Loading tokenizer...")
72
  tokenizer = M2M100Tokenizer.from_pretrained(
73
  pretrained_model_name_or_path=model_name, cache_dir=cache_dir
74
  )
75
- print("Loading model...")
76
  model = M2M100ForConditionalGeneration.from_pretrained(
77
  pretrained_model_name_or_path=model_name, cache_dir=cache_dir
78
  )
 
68
  mixed_precision=precision if precision != "32" else "no", split_batches=True
69
  )
70
 
71
+ print(f"Loading tokenizer {model_name}...")
72
  tokenizer = M2M100Tokenizer.from_pretrained(
73
  pretrained_model_name_or_path=model_name, cache_dir=cache_dir
74
  )
75
+ print(f"Loading model {model_name}...")
76
  model = M2M100ForConditionalGeneration.from_pretrained(
77
  pretrained_model_name_or_path=model_name, cache_dir=cache_dir
78
  )