Skip to content

Commit

Permalink
Informative (#35059)
Browse files Browse the repository at this point in the history
* fix

* fix

* fix

* fix

---------

Co-authored-by: ydshieh <[email protected]>
  • Loading branch information
ydshieh and ydshieh authored Dec 5, 2024
1 parent 1ed1de2 commit beb2c66
Show file tree
Hide file tree
Showing 4 changed files with 8 additions and 4 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -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(
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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."
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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."
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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",
Expand Down

0 comments on commit beb2c66

Please sign in to comment.