lee/BitNet
1
0
Fork 0
mirror of https://github.com/microsoft/BitNet.git synced 2026-08-25 18:36:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

add falcon-e support

Browse Source
This commit is contained in:
younesbelkada
2025-05-14 17:07:05 +04:00
parent c9e752c9d7
commit de371b708d
2 changed files with 10 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
setup_env.py
+7 -1
View File
@@ -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:
utils/convert-hf-to-gguf-bitnet.py
+3
View File
@@ -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"