Merge pull request #234 from Taik/main

This commit is contained in:
paul-gauthier 2023-09-08 17:07:02 -07:00 committed by GitHub
commit e073d6901b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 9 additions and 1 deletions

View file

@ -55,6 +55,7 @@ class Coder:
main_model, main_model,
edit_format, edit_format,
io, io,
skip_model_availabily_check=False,
**kwargs, **kwargs,
): ):
from . import EditBlockCoder, WholeFileCoder from . import EditBlockCoder, WholeFileCoder
@ -62,7 +63,7 @@ class Coder:
if not main_model: if not main_model:
main_model = models.GPT35_16k 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 not check_model_availability(main_model):
if main_model != models.GPT4: if main_model != models.GPT4:
io.tool_error( 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, default=models.GPT4.name,
help=f"Specify the model to use for the main chat (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( core_group.add_argument(
"-3", "-3",
action="store_const", action="store_const",
@ -465,6 +471,7 @@ def main(argv=None, input=None, output=None, force_git_root=None):
main_model, main_model,
args.edit_format, args.edit_format,
io, io,
args.skip_model_availability_check,
## ##
fnames=fnames, fnames=fnames,
git_dname=git_dname, git_dname=git_dname,