From 8b2b885e63a6a675fe34bb8c2d559cc7fd0ed61a Mon Sep 17 00:00:00 2001 From: Saifeddine ALOUI Date: Fri, 8 Dec 2023 00:59:46 +0100 Subject: [PATCH] Fixed for new gpt4all code --- lollms/binding.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lollms/binding.py b/lollms/binding.py index 3e10f57..ce0fb3f 100644 --- a/lollms/binding.py +++ b/lollms/binding.py @@ -437,7 +437,7 @@ class LLMBinding: """ models = [] for models_folder in self.models_folders: - if models_folder.name in ["ggml","gguf"]: + if models_folder.name in ["ggml","gguf","gpt4all"]: models+=[f.name for f in models_folder.iterdir() if f.suffix in self.supported_file_extensions or f.suffix==".reference"] else: models+=[f.name for f in models_folder.iterdir() if f.is_dir() and not f.stem.startswith(".") or f.suffix==".reference"]