remove CORRECTED llama 70b

e8e05e15a130aecc09d81581e9dc3d5ab2799bb4 · 2024-11-03 18:41:24 -0500 · Alex Cheema

Files touched

M exo/models.py

Diff

commit e8e05e15a130aecc09d81581e9dc3d5ab2799bb4
Author: Alex Cheema <alexcheema123@gmail.com>
Date:   Sun Nov 3 18:41:24 2024 -0500

    remove CORRECTED llama 70b
---
 exo/models.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/exo/models.py b/exo/models.py
index 5791bdd5..dce0693c 100644
--- a/exo/models.py
+++ b/exo/models.py
@@ -13,7 +13,7 @@ model_base_shards = {
     "TinygradDynamicShardInferenceEngine": Shard(model_id="NousResearch/Meta-Llama-3.1-70B-Instruct", start_layer=0, end_layer=0, n_layers=80),
   },
   "llama-3.1-70b-bf16": {
-    "MLXDynamicShardInferenceEngine": Shard(model_id="mlx-community/Meta-Llama-3.1-70B-Instruct-bf16-CORRECTED", start_layer=0, end_layer=0, n_layers=80),
+    "MLXDynamicShardInferenceEngine": Shard(model_id="mlx-community/Meta-Llama-3.1-70B-Instruct-bf16", start_layer=0, end_layer=0, n_layers=80),
     "TinygradDynamicShardInferenceEngine": Shard(model_id="NousResearch/Meta-Llama-3.1-70B-Instruct", start_layer=0, end_layer=0, n_layers=80),
   },
   "llama-3.1-405b": {"MLXDynamicShardInferenceEngine": Shard(model_id="mlx-community/Meta-Llama-3.1-405B-4bit", start_layer=0, end_layer=0, n_layers=126),},

← 436709e5 revert back to CORRECTED · back to Exo · revert back to CORRECTED 6ae6ebeb →