From beb2c66ec369135b4a41e8552948a0e7e073dceb Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Thu, 5 Dec 2024 09:50:27 +0100 Subject: [PATCH] Informative (#35059) * fix * fix * fix * fix --------- Co-authored-by: ydshieh --- .../convert_transfo_xl_original_tf_checkpoint_to_pytorch.py | 3 ++- .../models/maskformer/convert_maskformer_resnet_to_pytorch.py | 3 ++- .../models/maskformer/convert_maskformer_swin_to_pytorch.py | 3 ++- .../models/perceiver/convert_perceiver_haiku_to_pytorch.py | 3 ++- 4 files changed, 8 insertions(+), 4 deletions(-) diff --git a/src/transformers/models/deprecated/transfo_xl/convert_transfo_xl_original_tf_checkpoint_to_pytorch.py b/src/transformers/models/deprecated/transfo_xl/convert_transfo_xl_original_tf_checkpoint_to_pytorch.py index cf191f797f72df..2c7b687c4d9897 100644 --- a/src/transformers/models/deprecated/transfo_xl/convert_transfo_xl_original_tf_checkpoint_to_pytorch.py +++ b/src/transformers/models/deprecated/transfo_xl/convert_transfo_xl_original_tf_checkpoint_to_pytorch.py @@ -109,7 +109,8 @@ def convert_transfo_xl_checkpoint_to_pytorch( "--transfo_xl_dataset_file", default="", type=str, - help="An optional dataset file to be converted in a vocabulary.", + help="An optional dataset file to be converted in a vocabulary.\n" + "Given the files are in the pickle format, please be wary of passing it files you trust.", ) args = parser.parse_args() convert_transfo_xl_checkpoint_to_pytorch( diff --git a/src/transformers/models/maskformer/convert_maskformer_resnet_to_pytorch.py b/src/transformers/models/maskformer/convert_maskformer_resnet_to_pytorch.py index 34ac49403c95b1..3ca9d9dfc3d010 100644 --- a/src/transformers/models/maskformer/convert_maskformer_resnet_to_pytorch.py +++ b/src/transformers/models/maskformer/convert_maskformer_resnet_to_pytorch.py @@ -374,7 +374,8 @@ def convert_maskformer_checkpoint( "--checkpoint_path", type=str, required=True, - help=("Path to the original pickle file (.pkl) of the original checkpoint.",), + help="Path to the original pickle file (.pkl) of the original checkpoint.\n" + "Given the files are in the pickle format, please be wary of passing it files you trust.", ) parser.add_argument( "--pytorch_dump_folder_path", default=None, type=str, help="Path to the output PyTorch model directory." diff --git a/src/transformers/models/maskformer/convert_maskformer_swin_to_pytorch.py b/src/transformers/models/maskformer/convert_maskformer_swin_to_pytorch.py index 4917d97629bc06..41e8b4888810b3 100644 --- a/src/transformers/models/maskformer/convert_maskformer_swin_to_pytorch.py +++ b/src/transformers/models/maskformer/convert_maskformer_swin_to_pytorch.py @@ -317,7 +317,8 @@ def convert_maskformer_checkpoint( "--checkpoint_path", default="/Users/nielsrogge/Documents/MaskFormer_checkpoints/MaskFormer-Swin-tiny-ADE20k/model.pkl", type=str, - help="Path to the original state dict (.pth file).", + help="Path to the original state dict (.pth file).\n" + "Given the files are in the pickle format, please be wary of passing it files you trust.", ) parser.add_argument( "--pytorch_dump_folder_path", default=None, type=str, help="Path to the output PyTorch model directory." diff --git a/src/transformers/models/perceiver/convert_perceiver_haiku_to_pytorch.py b/src/transformers/models/perceiver/convert_perceiver_haiku_to_pytorch.py index 190b4f51f620ec..082b9449374a7e 100644 --- a/src/transformers/models/perceiver/convert_perceiver_haiku_to_pytorch.py +++ b/src/transformers/models/perceiver/convert_perceiver_haiku_to_pytorch.py @@ -444,7 +444,8 @@ def convert_perceiver_checkpoint(pickle_file, pytorch_dump_folder_path, architec type=str, default=None, required=True, - help="Path to local pickle file of a Perceiver checkpoint you'd like to convert.", + help="Path to local pickle file of a Perceiver checkpoint you'd like to convert.\n" + "Given the files are in the pickle format, please be wary of passing it files you trust.", ) parser.add_argument( "--pytorch_dump_folder_path",