SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

updating CLI

Browse Source
This commit is contained in:
thomwolf
2019-10-17 15:10:11 +02:00
committed by Morgan Funtowicz
parent 7c1697562a
commit 31a3a73ee3
6 changed files with 155 additions and 116 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
transformers-cli
View File

@@ -3,6 +3,8 @@ from argparse import ArgumentParser
from transformers.commands.serving import ServeCommand
from transformers.commands.user import UserCommands
from transformers.commands.train import TrainCommand
from transformers.commands.convert import ConvertCommand
if __name__ == '__main__':
parser = ArgumentParser('Transformers CLI tool', usage='transformers-cli <command> [<args>]')
@@ -11,6 +13,8 @@ if __name__ == '__main__':
# Register commands
ServeCommand.register_subcommand(commands_parser)
UserCommands.register_subcommand(commands_parser)
TrainCommand.register_subcommand(commands_parser)
ConvertCommand.register_subcommand(commands_parser)
# Let's go
args = parser.parse_args()