ExternalVendorCode/lollms
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms.git synced 2025-04-10 20:29:54 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
Saifeddine ALOUI 2024-02-29 19:32:21 +01:00
parent 1813e4505b
commit d2d039ac81
1 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
lollms/binding.py
View File

@ -120,7 +120,17 @@ class LLMBinding:
def searchModelPath(self, model_name:str):
model_path=self.searchModelFolder(model_name)
mp:Path = model_path/str(model_name).split("/")[-1]
mp:Path = None
for f in model_path.iterdir():
print(f)
a = model_name.lower()
b = f.name.lower()
if a in b :
mp = f
break
if not mp:
return None
# model_path/str(model_name).split("/")[-1]
if "ggml" in str(mp).lower() or "gguf" in str(mp).lower():
if mp.is_dir():
for f in mp.iterdir():