fixed error

This commit is contained in:
Saifeddine ALOUI 2023-10-20 21:18:16 +02:00
parent ec5b4bd018
commit 9eee751ccf
2 changed files with 6 additions and 2 deletions

View File

@ -208,8 +208,12 @@ class LLMBinding:
ASCIIColors.red(f"UnInstalling {self.binding_folder_name}") ASCIIColors.red(f"UnInstalling {self.binding_folder_name}")
ASCIIColors.blue("*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*") ASCIIColors.blue("*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*")
def searchModelParentFolder(self, model_name:str): def searchModelParentFolder(self, model_name:str, model_type=None):
model_path=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: for mn in self.models_folders:
if mn.name in model_name.lower(): if mn.name in model_name.lower():
model_path = mn model_path = mn

View File

@ -26,7 +26,7 @@ def get_all_files(path):
setuptools.setup( setuptools.setup(
name="lollms", name="lollms",
version="5.8.8", version="5.8.9",
author="Saifeddine ALOUI", author="Saifeddine ALOUI",
author_email="aloui.saifeddine@gmail.com", author_email="aloui.saifeddine@gmail.com",
description="A python library for AI personality definition", description="A python library for AI personality definition",