fead: add OpenChat3.6 8B

umbertogriffo · Jun 28, 2024 · 190380b · 190380b
1 parent 3c171c7
commit 190380b
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/chatbot/bot/model/model_settings.py b/chatbot/bot/model/model_settings.py
@@ -4,7 +4,7 @@
 from bot.model.settings.llama_3 import LlamaThreeSettings
 from bot.model.settings.mistral import MistralSettings
 from bot.model.settings.neural_beagle import NeuralBeagleSettings
-from bot.model.settings.openchat import OpenChatSettings
+from bot.model.settings.openchat import OpenChat35Settings, OpenChat36Settings
 from bot.model.settings.phi_3 import PhiThreeSettings
 from bot.model.settings.stablelm_zephyr import StableLMZephyrSettings
 from bot.model.settings.starling import StarlingSettings
@@ -16,7 +16,8 @@ class ModelType(Enum):
     MISTRAL = "mistral"
     DOLPHIN = "dolphin"
     STABLELM_ZEPHYR = "stablelm-zephyr"
-    OPENCHAT = "openchat"
+    OPENCHAT_3_5 = "openchat-3.5"
+    OPENCHAT_3_6 = "openchat-3.6"
     STARLING = "starling"
     NEURAL_BEAGLE = "neural-beagle"
     PHI_3 = "phi-3"
@@ -28,7 +29,8 @@ class ModelType(Enum):
     ModelType.MISTRAL.value: MistralSettings,
     ModelType.DOLPHIN.value: DolphinSettings,
     ModelType.STABLELM_ZEPHYR.value: StableLMZephyrSettings,
-    ModelType.OPENCHAT.value: OpenChatSettings,
+    ModelType.OPENCHAT_3_5.value: OpenChat35Settings,
+    ModelType.OPENCHAT_3_6.value: OpenChat36Settings,
     ModelType.STARLING.value: StarlingSettings,
     ModelType.NEURAL_BEAGLE.value: NeuralBeagleSettings,
     ModelType.PHI_3.value: PhiThreeSettings,