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": {