Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-02-24 04:01:10 +01:00
Code Issues Packages Projects Releases Wiki Activity

Fix intermixed missing in cli_argument_parser.py (#1437)

Browse Source 
Co-authored by: @yaqiangsun
This commit is contained in:
yaqiang.sun
2025-08-13 22:23:25 +08:00
committed by GitHub
parent 179b762254
commit 3797b4b8d1
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
modelscope/trainers/cli_argument_parser.py
View File

@@ -21,12 +21,17 @@ class CliArgumentParser(ArgumentParser):
def get_manual_args(self, args):
return [arg[2:] for arg in args if arg.startswith('--')]
def _parse_known_args(self, args: List = None, namespace=None):
def _parse_known_args(self,
args: List = None,
namespace=None,
*args_extra,
**kwargs):
self.model_id = namespace.model if namespace is not None else None
if '--model' in args:
self.model_id = args[args.index('--model') + 1]
self.manual_args = self.get_manual_args(args)
return super()._parse_known_args(args, namespace)
return super()._parse_known_args(args, namespace, *args_extra,
**kwargs)
def print_help(self, file=None):
return super().print_help(file)