gnh1201/welsonjs
1
0
Fork 0
mirror of https://github.com/gnh1201/welsonjs.git synced 2025-02-06 06:54:58 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update language-inference-engine.js

Browse Source
This commit is contained in:
Namhyeon Go 2025-01-30 19:47:30 +09:00
parent 408e98e46b
commit 3cf7f6c07c
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
lib/language-inference-engine.js
View File

@ -329,7 +329,9 @@ var ENGINE_PROFILES = {
"availableModels": [
"openchat-3.5-0106",
"qwen1.5-14b-chat-awq",
"gemma-7b-it"
"gemma-7b-it",
"una-cybertron-7b-v2-bf16",
"starling-lm-7b-beta"
],
"headers": {
"Content-Type": "application/json",
@ -373,12 +375,13 @@ function LanguageInferenceEngine() {
this.setProvider = function(provider) {
this.provider = provider;
if (this.provider in ENGINE_PROFILES) {
this.engineProfile = ENGINE_PROFILES[provider];
this.type = this.engineProfile.type;
this.model = this.engineProfile.availableModels[0];
}
return this;
};