Add option to skip model availability check

This commit is contained in:
Thinh Nguyen 2023-09-08 14:01:30 -07:00
parent 2a8c7ef70d
commit 45298f8af6
No known key found for this signature in database
2 changed files with 9 additions and 1 deletions

View file

@ -55,6 +55,7 @@ class Coder:
main_model,
edit_format,
io,
skip_model_availabily_check,
**kwargs,
):
from . import EditBlockCoder, WholeFileCoder
@ -62,7 +63,7 @@ class Coder:
if not main_model:
main_model = models.GPT35_16k
if not main_model.always_available:
if not skip_model_availabily_check and not main_model.always_available:
if not check_model_availability(main_model):
if main_model != models.GPT4:
io.tool_error(

View file

@ -142,6 +142,12 @@ def main(argv=None, input=None, output=None, force_git_root=None):
default=models.GPT4.name,
help=f"Specify the model to use for the main chat (default: {models.GPT4.name})",
)
core_group.add_argument(
"--skip-model-availability-check",
metavar="SKIP_MODEL_AVAILABILITY_CHECK",
default=False,
help="Override to skip model availability check (default: False)",
)
core_group.add_argument(
"-3",
action="store_const",
@ -465,6 +471,7 @@ def main(argv=None, input=None, output=None, force_git_root=None):
main_model,
args.edit_format,
io,
args.skip_model_availability_check,
##
fnames=fnames,
git_dname=git_dname,