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

output

Browse Source
This commit is contained in:
saloui 2023-05-26 16:26:16 +02:00
parent 7045d3bb23
commit 4d112f5576
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/__init__.py
View File

@ -201,7 +201,7 @@ class ModelProcess:
if self.config["model"] is None:
print("No model is selected.\nPlease select a backend and a model to start using the ui.")
else:
print("Couldn't build model")
print(f"Couldn't build model {self.config['model']} : {ex}")
self.model = None
self._set_config_result['status'] ='failed'
self._set_config_result['binding_status'] ='failed'

2
bindings/c_transformers/__init__.py
View File

@ -44,7 +44,7 @@ class CTRansformers(LLMBinding):
model_type='dolly-v2'
elif 'starcoder' in self.config['model']:
model_type='starcoder'
elif 'llama' in self.config['model'] or 'wizardLM' in self.config['model']:
elif 'llama' in self.config['model'].lower() or 'wizardlm' in self.config['model'].lower() or 'vigogne' in self.config['model'].lower():
model_type='llama'
elif 'mpt' in self.config['model']:
model_type='mpt'