ExternalVendorCode/lollms-webui
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms-webui.git synced 2025-02-21 01:31:20 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

bugfix

Browse Source
This commit is contained in:
Saifeddine ALOUI 2023-06-10 02:22:49 +02:00
parent a57bf5f3f2
commit ce25ae011e
2 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
api/__init__.py
View File

@ -253,7 +253,7 @@ class ModelProcess:
for personality in self.config['personalities']:
try:
print(f" {personality}")
personality_path = lollms_path/f"personalities_zoo/{personality}"
personality_path = lollms_personalities_zoo_path/f"{personality}"
personality = AIPersonality(personality_path, run_scripts=False)
mounted_personalities.append(personality)
except Exception as ex:
@ -293,7 +293,7 @@ class ModelProcess:
self._set_config_result['status'] ='semi_failed'
self._set_config_result['personalities_status'] ='semi_failed'
self.personality = self.config["personalities"][self.config['active_personality_id']]
self.personality = self.mounted_personalities[self.config['active_personality_id']]
self.mounted_personalities = self.config["personalities"]
print("Personality set successfully")
@ -349,6 +349,7 @@ class ModelProcess:
self.start_signal.clear()
print("Finished executing the generation")
except Exception as ex:
print("Couldn't start generation")
print(ex)
time.sleep(1)
def _generate(self, prompt, n_predict=50, callback=None):
@ -466,10 +467,9 @@ class LoLLMsAPPI():
#Create and launch the process
self.process = ModelProcess(config)
self.config = config
self.mounted_personalities = self.config["personalities"]
self.binding = self.process.rebuild_binding(self.config)
self.personalities = self.process.rebuild_personalities()
self.personality = self.personalities[self.config["active_personality_id"]]
self.mounted_personalities = self.process.rebuild_personalities()
self.personality = self.mounted_personalities[self.config["active_personality_id"]]
if config["debug"]:
print(print(f"{self.personality}"))
self.config_file_path = config_file_path

6
app.py
View File

@ -714,7 +714,7 @@ class LoLLMsWebUI(LoLLMsAPPI):
if config_file.exists():
self.config["personalities"].append(package_path)
self.personalities = self.process.rebuild_personalities()
self.personality = self.personalities[self.config["active_personality_id"]]
self.personality = self.mounted_personalities[self.config["active_personality_id"]]
self.apply_settings()
return jsonify({"status": True,
"personalities":self.config["personalities"],
@ -742,7 +742,7 @@ class LoLLMsWebUI(LoLLMsAPPI):
self.config["active_personality_id"]=0
if len(self.config["personalities"])>0:
self.personalities = self.process.rebuild_personalities()
self.personality = self.personalities[self.config["active_personality_id"]]
self.personality = self.mounted_personalities[self.config["active_personality_id"]]
else:
self.personalities = []
self.personality = None
@ -759,7 +759,7 @@ class LoLLMsWebUI(LoLLMsAPPI):
id = request.files['id']
if id<len(self.config["personalities"]):
self.config["active_personality_id"]=id
self.personality = self.personalities[self.config["active_personality_id"]]
self.personality = self.mounted_personalities[self.config["active_personality_id"]]
self.apply_settings()
return jsonify({"status": True})
else: