ExternalVendorCode/lollms
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms.git synced 2025-02-23 18:40:16 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update binding.py

Browse Source
This commit is contained in:
Saifeddine ALOUI 2023-10-19 09:24:39 +02:00 committed by GitHub
parent fc50cd81c6
commit 54fa602162
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lollms/binding.py
View File

@ -225,11 +225,11 @@ class LLMBinding:
if mn.name in model_name.lower(): if mn.name in model_name.lower():
if mn.name == "ggml": if mn.name == "ggml":
idx = model_name.index("-GGML") idx = model_name.index("-GGML")
models=[m for m in mn.iterdir() if model_name[:idx] in m.name] models=[m for m in mn.iterdir() if model_name[:idx].lower() in m.name.lower()]
model_path = mn/models[0].name model_path = mn/models[0].name
elif mn.name == "gguf": elif mn.name == "gguf":
idx = model_name.index("-GGUF") idx = model_name.index("-GGUF")
models=[m for m in mn.iterdir() if model_name[:idx] in m.name] models=[m for m in mn.iterdir() if model_name[:idx].lower() in m.name.lower()]
model_path = mn/models[0].name model_path = mn/models[0].name
else: else:
model_path = mn/model_name model_path = mn/model_name