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 |
)
|