| 1234567891011121314151617181920212223242526272829303132333435 |
- # Copyright (c) Alibaba, Inc. and its affiliates.
- from argparse import ArgumentParser
- from modelscope.cli.base import CLICommand
- from modelscope.hub.api import HubApi
- def subparser_func(args):
- """ Function which will be called for a specific sub parser.
- """
- return LoginCMD(args)
- class LoginCMD(CLICommand):
- name = 'login'
- def __init__(self, args):
- self.args = args
- @staticmethod
- def define_args(parsers: ArgumentParser):
- """ define args for login command.
- """
- parser = parsers.add_parser(LoginCMD.name)
- parser.add_argument(
- '--token',
- type=str,
- required=True,
- help='The Access Token for modelscope.')
- parser.set_defaults(func=subparser_func)
- def execute(self):
- api = HubApi()
- api.login(self.args.token)
|
