Support --model as alias for --model-path in CLI (#23894)
This commit is contained in:
@@ -103,11 +103,11 @@ def get_model_path(extra_argv):
|
|||||||
# Find the model_path argument
|
# Find the model_path argument
|
||||||
model_path = None
|
model_path = None
|
||||||
for i, arg in enumerate(extra_argv):
|
for i, arg in enumerate(extra_argv):
|
||||||
if arg == "--model-path":
|
if arg in ("--model-path", "--model"):
|
||||||
if i + 1 < len(extra_argv):
|
if i + 1 < len(extra_argv):
|
||||||
model_path = extra_argv[i + 1]
|
model_path = extra_argv[i + 1]
|
||||||
break
|
break
|
||||||
elif arg.startswith("--model-path="):
|
elif arg.startswith("--model-path=") or arg.startswith("--model="):
|
||||||
model_path = arg.split("=", 1)[1]
|
model_path = arg.split("=", 1)[1]
|
||||||
break
|
break
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user