diff --git a/src/transformers/models/marian/convert_marian_to_pytorch.py b/src/transformers/models/marian/convert_marian_to_pytorch.py index f086e480dfffdc..3938bc794571b4 100644 --- a/src/transformers/models/marian/convert_marian_to_pytorch.py +++ b/src/transformers/models/marian/convert_marian_to_pytorch.py @@ -701,7 +701,12 @@ def unzip(zip_path: str, dest_dir: str) -> None: """ parser = argparse.ArgumentParser() # Required parameters - parser.add_argument("--src", type=str, help="path to marian model sub dir", default="en-de") + parser.add_argument( + "--src", + type=str, + help="path to marian model sub dir. yaml.load will be used to load the configuration file, please be weary of which file you're loading.", + default="en-de", + ) parser.add_argument("--dest", type=str, default=None, help="Path to the output PyTorch model.") args = parser.parse_args() diff --git a/src/transformers/models/mobilevitv2/convert_mlcvnets_to_pytorch.py b/src/transformers/models/mobilevitv2/convert_mlcvnets_to_pytorch.py index 518dc949a47b96..e0aed91cd28977 100644 --- a/src/transformers/models/mobilevitv2/convert_mlcvnets_to_pytorch.py +++ b/src/transformers/models/mobilevitv2/convert_mlcvnets_to_pytorch.py @@ -314,7 +314,12 @@ def convert_mobilevitv2_checkpoint(task_name, checkpoint_path, orig_config_path, parser.add_argument( "--orig_checkpoint_path", required=True, type=str, help="Path to the original state dict (.pt file)." ) - parser.add_argument("--orig_config_path", required=True, type=str, help="Path to the original config file.") + parser.add_argument( + "--orig_config_path", + required=True, + type=str, + help="Path to the original config file. yaml.load will be used to load the file, please be weary of which file you're loading.", + ) parser.add_argument( "--pytorch_dump_folder_path", required=True, type=str, help="Path to the output PyTorch model directory." )