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