gpt4all.py: GGUF

This commit is contained in:
Cebtenzzre 2023-09-26 09:30:22 -04:00 committed by Adam Treat
parent 1d29e4696c
commit 34f2ec2b33

View File

@ -423,6 +423,6 @@ def empty_chat_session(system_prompt: str = "") -> List[MessageType]:
def append_bin_suffix_if_missing(model_name):
if not model_name.endswith(".bin"):
if not model_name.endswith((".bin", ".gguf")):
model_name += ".bin"
return model_name