fix: simplify naming

kaito-project · Jan 30, 2024 · d32364c · d32364c
1 parent 4dc7ea7
commit d32364c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/.github/workflows/e2e-preset-configs.json b/.github/workflows/e2e-preset-configs.json
@@ -26,13 +26,13 @@
         "node-osdisk-size": 400
       },
       {
-        "name": "mistral-7b-v0.1",
+        "name": "mistral-7b",
         "node-count": 1,
         "node-vm-size": "Standard_NC12s_v3",
         "node-osdisk-size": 100
       },
       {
-        "name": "mistral-7b-instruct-v0.1",
+        "name": "mistral-7b-instruct",
         "node-count": 1,
         "node-vm-size": "Standard_NC12s_v3",
         "node-osdisk-size": 100

diff --git a/presets/models/supported_models.yaml b/presets/models/supported_models.yaml
@@ -48,12 +48,12 @@ models:
     tag: 0.0.1
 
   # Mistral
-  - name: mistral-7b-v0.1
+  - name: mistral-7b
     type: text-generation 
     version: https://huggingface.co/mistralai/Mistral-7B-v0.1/commit/26bca36bde8333b5d7f72e9ed20ccda6a618af24
     runtime: tfs
     tag: 0.0.1
-  - name: mistral-7b-instruct-v0.1
+  - name: mistral-7b-instruct
     type: text-generation
     version: https://huggingface.co/mistralai/Mistral-7B-Instruct-v0.1/commit/9ab9e76e2b09f9f29ea2d56aa5bd139e4445c59e
     runtime: tfs