Binding pipelines to the cli.

This commit is contained in:
Morgan Funtowicz committed 2019-12-15 01:37:16 +01:00
1 parent 0b51532ce9
commit f1971bf303
3 files changed
+152 -10

No files matched your search

+3 -1
View File
@@ -2,6 +2,7 @@
from argparse import ArgumentParser
from transformers.commands.download import DownloadCommand
from transformers.commands.run import RunCommand
from transformers.commands.serving import ServeCommand
from transformers.commands.user import UserCommands
from transformers.commands.train import TrainCommand
@@ -14,9 +15,10 @@ if __name__ == '__main__':
# Register commands
ConvertCommand.register_subcommand(commands_parser)
DownloadCommand.register_subcommand(commands_parser)
RunCommand.register_subcommand(commands_parser)
ServeCommand.register_subcommand(commands_parser)
UserCommands.register_subcommand(commands_parser)
TrainCommand.register_subcommand(commands_parser)
UserCommands.register_subcommand(commands_parser)
# Let's go
args = parser.parse_args()