From 54fa602162552c6ec73ccc3bfb6ee1a4ba6d99b8 Mon Sep 17 00:00:00 2001 From: Saifeddine ALOUI Date: Thu, 19 Oct 2023 09:24:39 +0200 Subject: [PATCH] Update binding.py --- lollms/binding.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/lollms/binding.py b/lollms/binding.py index 9101d83..8e31485 100644 --- a/lollms/binding.py +++ b/lollms/binding.py @@ -225,11 +225,11 @@ class LLMBinding: if mn.name in model_name.lower(): if mn.name == "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 elif mn.name == "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 else: model_path = mn/model_name