Support GIST-small-Embedding-v0 embedding model

jncraton · Apr 20, 2024 · 39a6210 · 39a6210
1 parent 8955007
commit 39a6210
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/changelog.md b/changelog.md
@@ -4,6 +4,7 @@
 
 ### Added
 
+- Support GIST-small-Embedding-v0 embedding model
 - Store model runtime stats to improve benchmarking and analysis
 
 ## 0.19 - 2024-04-18

diff --git a/languagemodels/config.py b/languagemodels/config.py
@@ -373,6 +373,15 @@ class ModelFilterException(Exception):
         "architecture": "encoder-only-transformer",
         "license": "mit",
     },
+    {
+        "name": "GIST-small-Embedding-v0",
+        "tuning": "embedding",
+        "params": 33e6,
+        "quantization": "int8",
+        "backend": "ct2",
+        "architecture": "encoder-only-transformer",
+        "license": "mit",
+    },
     {
         "name": "bge-small-en",
         "tuning": "embedding",