diff --git a/setup_env.py b/setup_env.py
index f15d65f4..09385a91 100644
--- a/setup_env.py
+++ b/setup_env.py
@@ -23,19 +23,19 @@
     "tiiuae/Falcon3-7B-Instruct-1.58bit": {
         "model_name": "Falcon3-7B-Instruct-1.58bit",
     },
-    "tiiuae/Falcon3-7B-1.58bit": {
+    "tiiuae/Falcon3-7B-Base-1.58bit": {
         "model_name": "Falcon3-7B-1.58bit",
     },
     "tiiuae/Falcon3-10B-Instruct-1.58bit": {
         "model_name": "Falcon3-10B-Instruct-1.58bit",
     },
-    "tiiuae/Falcon3-10B-1.58bit": {
+    "tiiuae/Falcon3-10B-Base-1.58bit": {
         "model_name": "Falcon3-10B-1.58bit",
     },
     "tiiuae/Falcon3-3B-Instruct-1.58bit": {
         "model_name": "Falcon3-3B-Instruct-1.58bit",
     },
-    "tiiuae/Falcon3-3B-1.58bit": {
+    "tiiuae/Falcon3-3B-Base-1.58bit": {
         "model_name": "Falcon3-3B-1.58bit",
     },
     "tiiuae/Falcon3-1B-Instruct-1.58bit": {