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(