ExternalVendorCode/lollms
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms.git synced 2025-01-20 03:36:30 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

upgraded binding

Browse Source
This commit is contained in:
Saifeddine ALOUI 2024-01-21 23:38:25 +01:00
parent 00d593815e
commit 2821ef2319
1 changed files with 6 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
lollms/binding.py
View File

@ -529,25 +529,11 @@ class LLMBinding:
def searchModelPath(self, model_name:str):
model_path=None
for mn in self.models_folders:
if mn.name in model_name.lower():
if mn.name == "ggml":
try:
idx = model_name.index("-GGML")
models=[m for m in mn.iterdir() if model_name[:idx].lower() in m.name.lower()]
model_path = mn/models[0].name
except:
model_path = mn/model_name
elif mn.name == "gguf":
try:
idx = model_name.index("-GGUF")
models=[m for m in mn.iterdir() if model_name[:idx].lower() in m.name.lower()]
model_path = mn/models[0].name
except:
model_path = mn/model_name
else:
model_path = mn/model_name
break
if model_path is None:
for f in mn.iterdir():
if model_name.lower().replace("-gguf","").replace("-ggml","") in str(f).lower():
return f
model_path = self.models_folders[0]/model_name
return model_path