diff --git a/class.py b/class.py index ce9d56e2a..1af9082a1 100644 --- a/class.py +++ b/class.py @@ -41,7 +41,7 @@ class model_backend(InferenceModel): foundfile = False try: files = os.listdir(model_path) - foundfile = len([filename for filename in files if (("ggml" in filename.lower() and ".bin" in filename.lower()) or ".gguf" in filename.lower())])>1 + foundfile = len([filename for filename in files if (("ggml" in filename.lower() and ".bin" in filename.lower()) or ".gguf" in filename.lower())])>0 except: pass return foundfile