chore: add Yi-large

namwoam · Aug 9, 2024 · 27b3a0a · 27b3a0a
1 parent 09ca941
commit 27b3a0a
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/ai/fireworksai/v0/config/tasks.json b/ai/fireworksai/v0/config/tasks.json
@@ -162,7 +162,8 @@
             "starcoder-7b",
             "phi-3-vision-128k-instruct",
             "qwen2-72b-instruct",
-            "mythomax-l2-13b"
+            "mythomax-l2-13b",
+            "yi-large"
           ],
           "example": "llama-v3p1-8b-instruct",
           "description": "The OSS model to be used",
@@ -190,7 +191,8 @@
               "starcoder-7b",
               "phi-3-vision-128k-instruct",
               "qwen2-72b-instruct",
-              "mythomax-l2-13b"
+              "mythomax-l2-13b",
+              "yi-large"
             ],
             "targets": [
               "setup.api-key"