ExternalVendorCode/lollms
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms.git synced 2025-01-18 18:56:28 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fixed smart router

Browse Source
This commit is contained in:
Saifeddine ALOUI 2024-11-20 01:12:15 +01:00
parent 4df0a41564
commit a2408dc1aa
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
lollms/app.py
View File

@ -211,13 +211,13 @@ class LollmsApplication(LoLLMsCom):
model_name = parts[1]
return binding, model_name
def select_model(self, binding_name, model_name):
def select_model(self, binding_name, model_name, destroy_previous_model=True):
self.config["binding_name"] = binding_name
self.config["model_name"] = model_name
print(f"New binding selected : {binding_name}")
try:
if self.binding:
if self.binding and destroy_previous_model:
self.binding.destroy_model()
self.binding = None
self.model = None
@ -244,6 +244,10 @@ class LollmsApplication(LoLLMsCom):
print(f"New model active : {model.model_name}")
self.model = model
self.binding = model
self.personality.model = model
for per in self.mounted_personalities:
if per is not None:
per.model = self.model
self.config["binding_name"] = model.binding_folder_name
self.config["model_name"] = model.model_name