feat: add phi-2

kaito-project · Feb 7, 2024 · 152af3f · 152af3f
1 parent 61e7565
commit 152af3f
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/presets/models/supported_models.yaml b/presets/models/supported_models.yaml
@@ -58,3 +58,10 @@ models:
     version: https://huggingface.co/mistralai/Mistral-7B-Instruct-v0.2/commit/b70aa86578567ba3301b21c8a27bea4e8f6d6d61
     runtime: tfs
     tag: 0.0.2
+
+  # Phi-2
+  - name: phi-2
+    type: text-generation 
+    version: https://huggingface.co/microsoft/phi-2/commit/b10c3eba545ad279e7208ee3a5d644566f001670
+    runtime: tfs
+    tag: 0.0.1