From c96c2499d6ff6422c62c6409141896fc86749b92 Mon Sep 17 00:00:00 2001
From: potassiummmm <54800242+potassiummmm@users.noreply.github.com>
Date: Thu, 19 Dec 2024 01:20:14 +0800
Subject: [PATCH] Fix model name in setup_env.py

---
 setup_env.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup_env.py b/setup_env.py
index 4b38013..9256324 100644
--- a/setup_env.py
+++ b/setup_env.py
@@ -21,7 +21,7 @@ SUPPORTED_HF_MODELS = {
         "model_name": "Llama3-8B-1.58-100B-tokens",
     },
     "tiiuae/Falcon3-7B-Instruct-1.58bit": {
-        "model_name": "Falcon3-7B-1.58bit",
+        "model_name": "Falcon3-7B-Instruct-1.58bit",
     },
     "tiiuae/Falcon3-7B-1.58bit": {
         "model_name": "Falcon3-7B-1.58bit",