From de371b708d5f1c3923812f4838581e0f81439436 Mon Sep 17 00:00:00 2001
From: younesbelkada <younesbelkada@gmail.com>
Date: Wed, 14 May 2025 17:07:05 +0400
Subject: [PATCH] add falcon-e support

---
 setup_env.py                       | 8 +++++++-
 utils/convert-hf-to-gguf-bitnet.py | 3 +++
 2 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/setup_env.py b/setup_env.py
index dfad6c3..2544c51 100644
--- a/setup_env.py
+++ b/setup_env.py
@@ -44,6 +44,12 @@ SUPPORTED_HF_MODELS = {
     "microsoft/BitNet-b1.58-2B-4T": {
         "model_name": "BitNet-b1.58-2B-4T",
     },
+    "tiiuae/Falcon-E-3B-Instruct": {
+        "model_name": "Falcon-E-3B-Instruct",
+    },
+    "tiiuae/Falcon-E-1B-Instruct": {
+        "model_name": "Falcon-E-1B-Instruct",
+    },
 }
 
 SUPPORTED_QUANT_TYPES = {
@@ -144,7 +150,7 @@ def setup_gguf():
 def gen_code():
     _, arch = system_info()
     
-    llama3_f3_models = set([model['model_name'] for model in SUPPORTED_HF_MODELS.values() if model['model_name'].startswith("Falcon3") or model['model_name'].startswith("Llama")])
+    llama3_f3_models = set([model['model_name'] for model in SUPPORTED_HF_MODELS.values() if model['model_name'].startswith("Falcon") or model['model_name'].startswith("Llama")])
 
     if arch == "arm64":
         if args.use_pretuned:
diff --git a/utils/convert-hf-to-gguf-bitnet.py b/utils/convert-hf-to-gguf-bitnet.py
index f525f58..23e8438 100644
--- a/utils/convert-hf-to-gguf-bitnet.py
+++ b/utils/convert-hf-to-gguf-bitnet.py
@@ -319,6 +319,9 @@ class Model(ABC):
         if chkhsh == "8aeee3860c56296a157a1fe2fad249ec40aa59b1bb5709f4ade11c4e6fe652ed":
             # ref: https://huggingface.co/tiiuae/falcon-7b
             res = "falcon"
+        if chkhsh == "a6b57017d60e6edb4d88ecc2845188e0eb333a70357e45dcc9b53964a73bbae6":
+            # ref: https://huggingface.co/tiiuae/Falcon-E-3B-Instruct
+            res = "falcon_e"
         if chkhsh == "0876d13b50744004aa9aeae05e7b0647eac9d801b5ba4668afc01e709c15e19f":
             # ref: https://huggingface.co/BAAI/bge-small-en-v1.5
             res = "bert-bge"