From 9eee751ccfe6acc0e71905d038adf0f2cc4e7e76 Mon Sep 17 00:00:00 2001 From: Saifeddine ALOUI Date: Fri, 20 Oct 2023 21:18:16 +0200 Subject: [PATCH] fixed error --- lollms/binding.py | 6 +++++- setup.py | 2 +- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/lollms/binding.py b/lollms/binding.py index 8e31485..19846bc 100644 --- a/lollms/binding.py +++ b/lollms/binding.py @@ -208,8 +208,12 @@ class LLMBinding: ASCIIColors.red(f"UnInstalling {self.binding_folder_name}") ASCIIColors.blue("*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*") - def searchModelParentFolder(self, model_name:str): + def searchModelParentFolder(self, model_name:str, model_type=None): model_path=None + if model_type is not None: + for mn in self.models_folders: + if mn.name.lower() == model_type.lower(): + return mn for mn in self.models_folders: if mn.name in model_name.lower(): model_path = mn diff --git a/setup.py b/setup.py index 120157f..0a7790a 100644 --- a/setup.py +++ b/setup.py @@ -26,7 +26,7 @@ def get_all_files(path): setuptools.setup( name="lollms", - version="5.8.8", + version="5.8.9", author="Saifeddine ALOUI", author_email="aloui.saifeddine@gmail.com", description="A python library for AI personality definition",