ExternalVendorCode/lollms
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms.git synced 2024-12-19 04:37:54 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fixed for new gpt4all code

Browse Source
This commit is contained in:
Saifeddine ALOUI 2023-12-08 00:59:46 +01:00
parent 3b8ca253ad
commit 8b2b885e63
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lollms/binding.py
View File

@ -437,7 +437,7 @@ class LLMBinding:
""" """
models = [] models = []
for models_folder in self.models_folders: 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"] models+=[f.name for f in models_folder.iterdir() if f.suffix in self.supported_file_extensions or f.suffix==".reference"]
else: else:
models+=[f.name for f in models_folder.iterdir() if f.is_dir() and not f.stem.startswith(".") or f.suffix==".reference"] models+=[f.name for f in models_folder.iterdir() if f.is_dir() and not f.stem.startswith(".") or f.suffix==".reference"]