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

Merge branch 'main' into small-fixes

Browse Source
This commit is contained in:
AndzejsP 2023-05-24 17:44:51 +03:00
commit c4cb9dc9bd
2 changed files with 17 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
app.py
View File

@ -683,16 +683,19 @@ class Gpt4AllWebUI(GPT4AllAPI):
print("New backend selected")
self.config['backend'] = backend
backend_ =self.process.load_backend(config["backend"])
models = backend_.list_models(self.config)
if len(models)>0:
self.backend = backend_
self.config['model'] = models[0]
# Build chatbot
return jsonify(self.process.set_config(self.config))
else:
return jsonify({"status": "no_models_found"})
try:
backend_ =self.process.load_backend(config["backend"],True)
models = backend_.list_models(self.config)
if len(models)>0:
self.backend = backend_
self.config['model'] = models[0]
# Build chatbot
return jsonify(self.process.set_config(self.config))
else:
return jsonify({"status": "no_models_found"})
except :
return jsonify({"status": "failed"})
return jsonify({"status": "error"})
def set_model(self):

4
gpt4all_api/api.py
View File

@ -155,6 +155,10 @@ class ModelProcess:
self.model = model_path
def set_config(self, config):
try:
self.set_config_result_queue.get_nowait()
except:
pass
self.set_config_queue.put(config)
# Wait for it t o be consumed
while self.set_config_result_queue.empty():