From d2d039ac81410a966dc24f041ee2c680bfba4f3c Mon Sep 17 00:00:00 2001 From: Saifeddine ALOUI Date: Thu, 29 Feb 2024 19:32:21 +0100 Subject: [PATCH] fix --- lollms/binding.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/lollms/binding.py b/lollms/binding.py index ebde503..15da025 100644 --- a/lollms/binding.py +++ b/lollms/binding.py @@ -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():