|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
import warnings |
|
|
|
|
|
from transformers import HfArgumentParser |
|
|
from transformers.commands.add_fast_image_processor import AddFastImageProcessorCommand |
|
|
from transformers.commands.add_new_model_like import AddNewModelLikeCommand |
|
|
from transformers.commands.chat import ChatCommand |
|
|
from transformers.commands.convert import ConvertCommand |
|
|
from transformers.commands.download import DownloadCommand |
|
|
from transformers.commands.env import EnvironmentCommand |
|
|
from transformers.commands.run import RunCommand |
|
|
from transformers.commands.serving import ServeCommand |
|
|
|
|
|
|
|
|
def main_cli(): |
|
|
warnings.warn( |
|
|
"`transformers-cli` is deprecated in favour of `transformers` directly and will be removed in v5.", |
|
|
DeprecationWarning, |
|
|
) |
|
|
main() |
|
|
|
|
|
|
|
|
def main(): |
|
|
parser = HfArgumentParser(prog="Transformers CLI tool", usage="transformers <command> [<args>]") |
|
|
commands_parser = parser.add_subparsers(help="transformers command helpers") |
|
|
|
|
|
|
|
|
ChatCommand.register_subcommand(commands_parser) |
|
|
ConvertCommand.register_subcommand(commands_parser) |
|
|
DownloadCommand.register_subcommand(commands_parser) |
|
|
EnvironmentCommand.register_subcommand(commands_parser) |
|
|
RunCommand.register_subcommand(commands_parser) |
|
|
ServeCommand.register_subcommand(commands_parser) |
|
|
AddNewModelLikeCommand.register_subcommand(commands_parser) |
|
|
AddFastImageProcessorCommand.register_subcommand(commands_parser) |
|
|
|
|
|
|
|
|
args = parser.parse_args() |
|
|
|
|
|
if not hasattr(args, "func"): |
|
|
parser.print_help() |
|
|
exit(1) |
|
|
|
|
|
|
|
|
service = args.func(args) |
|
|
service.run() |
|
|
|
|
|
|
|
|
if __name__ == "__main__": |
|
|
main() |
|
|
|