ExternalVendorCode/lollms
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms.git synced 2025-04-03 00:51:13 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fixed error

Browse Source
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

6
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

2
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",