diff --git a/lib/language-inference-engine.js b/lib/language-inference-engine.js index bed16ad..36db16d 100644 --- a/lib/language-inference-engine.js +++ b/lib/language-inference-engine.js @@ -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; };