diff --git a/setup_env.py b/setup_env.py
index 2544c51..f15d65f 100644
--- a/setup_env.py
+++ b/setup_env.py
@@ -50,6 +50,12 @@ SUPPORTED_HF_MODELS = {
     "tiiuae/Falcon-E-1B-Instruct": {
         "model_name": "Falcon-E-1B-Instruct",
     },
+    "tiiuae/Falcon-E-3B-Base": {
+        "model_name": "Falcon-E-3B-Base",
+    },
+    "tiiuae/Falcon-E-1B-Base": {
+        "model_name": "Falcon-E-1B-Base",
+    },
 }
 
 SUPPORTED_QUANT_TYPES = {