Spaces:
Running
Running
Print model name
Browse files- 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 |
)
|