Register mistral tiny (#123)

CHANGELOG.md

@@ -7,6 +7,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 ## [Unreleased]
 
 ### Added
+- Adds references to `mistral-tiny` (7bn parameter model from MistralAI) to the model registry for completeness.
 
 ### Fixed
 

Project.toml

@@ -1,7 +1,7 @@
 name = "PromptingTools"
 uuid = "670122d1-24a8-4d70-bfce-740807c42192"
 authors = ["J S @svilupp and contributors"]
-version = "0.18.0"
+version = "0.18.1"
 
 [deps]
 AbstractTrees = "1520ce14-60c1-5f80-bbc7-55ef81b5835c"

src/user_preferences.jl

@@ -306,9 +306,11 @@ aliases = merge(
         ## t-mixtral -> Together.ai Mixtral
         "tmixtral" => "mistralai/Mixtral-8x7B-Instruct-v0.1",
         ## Mistral AI
+        "mistral-tiny" => "mistral-tiny",
         "mistral-small" => "mistral-small-latest",
         "mistral-medium" => "mistral-medium-latest",
         "mistral-large" => "mistral-large-latest",
+        "mistralt" => "mistral-tiny",
         "mistrals" => "mistral-small-latest",
         "mistralm" => "mistral-medium-latest",
         "mistrall" => "mistral-large-latest",
@@ -435,6 +437,16 @@ registry = Dict{String, ModelSpec}(
         7e-7,
         7e-7,
         "Mistral AI's hosted version of openly available Mixtral-8x7B-v0.1. Good for more complicated tasks."),
+    "mistral-tiny" => ModelSpec("mistral-tiny",
+        MistralOpenAISchema(),
+        2e-6,
+        6e-6,
+        "Mistral AI's own finetune of their 7b model."),
+    "mistral-tiny-2312" => ModelSpec("mistral-tiny-2312",
+        MistralOpenAISchema(),
+        2e-6,
+        6e-6,
+        "Mistral AI's own finetune of their 7b model. Version 2312."),
     "mistral-small-latest" => ModelSpec("mistral-small-latest",
         MistralOpenAISchema(),
         2e-6,