baalajimaestro/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Fix boolean cli flags

Browse source
This commit is contained in:
Andrei Betlen 2023-09-13 22:56:10 -04:00
parent 83764c5aee
commit c999325e8e
1 changed files with 25 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
llama_cpp/server/__main__.py
View file

@ -58,13 +58,31 @@ if __name__ == "__main__":
description = field.description
if field.default is not None and description is not None:
description += f" (default: {field.default})"
base_type = get_base_type(field.annotation) if field.annotation is not None else str
list_type = contains_list_type(field.annotation)
if base_type is not bool:
parser.add_argument(
f"--{name}",
dest=name,
nargs="*" if contains_list_type(field.annotation) else None,
type=get_base_type(field.annotation) if field.annotation is not None else str,
nargs="*" if list_type else None,
type=base_type,
help=description,
)
if base_type is bool:
parser.add_argument(
f"--{name}",
dest=name,
action="store_true",
help=f"Disable {description}",
default=field.default,
)
parser.add_argument(
f"--no-{name}",
dest=name,
action="store_false",
help=f"Disable {description}",
default=field.default,
)
args = parser.parse_args()
settings = Settings(**{k: v for k, v in vars(args).items() if v is not None})