add -y to bypass prompt for transformers-cli upload (#7035)

This commit is contained in:
Stas Bekman
2020-09-10 01:58:29 -07:00
committed by GitHub
parent 76818cc4c6
commit 4ee1053dcf
2 changed files with 13 additions and 4 deletions

View File

@@ -136,6 +136,13 @@ Then log in using the same credentials as on huggingface.co. To upload your mode
This will upload the folder containing the weights, tokenizer and configuration we prepared in the previous section. This will upload the folder containing the weights, tokenizer and configuration we prepared in the previous section.
By default you will be prompted to confirm that you want these files to be uploaded. If you are uploading multiple models and need to script that process, you can add `-y` to bypass the prompt. For example:
::
transformers-cli upload -y path/to/awesome-name-you-picked/
If you want to upload a single file (a new version of your model, or the other framework checkpoint you want to add), If you want to upload a single file (a new version of your model, or the other framework checkpoint you want to add),
just type: just type:

View File

@@ -40,6 +40,7 @@ class UserCommands(BaseTransformersCLICommand):
upload_parser.add_argument( upload_parser.add_argument(
"--filename", type=str, default=None, help="Optional: override individual object filename on S3." "--filename", type=str, default=None, help="Optional: override individual object filename on S3."
) )
upload_parser.add_argument("-y", "--yes", action="store_true", help="Optional: answer Yes to the prompt")
upload_parser.set_defaults(func=lambda args: UploadCommand(args)) upload_parser.set_defaults(func=lambda args: UploadCommand(args))
@@ -221,10 +222,11 @@ class UploadCommand(BaseUserCommand):
) )
) )
choice = input("Proceed? [Y/n] ").lower() if not self.args.yes:
if not (choice == "" or choice == "y" or choice == "yes"): choice = input("Proceed? [Y/n] ").lower()
print("Abort") if not (choice == "" or choice == "y" or choice == "yes"):
exit() print("Abort")
exit()
print(ANSI.bold("Uploading... This might take a while if files are large")) print(ANSI.bold("Uploading... This might take a while if files are large"))
for filepath, filename in files: for filepath, filename in files:
try: try: