mirror of
https://github.com/unslothai/unsloth
synced 2026-04-21 13:37:39 +00:00
Revert "updated models template mappers. added lfm2.5vl450m to transformers 5…" (#4945)
This reverts commit bcf4fd6bd3.
This commit is contained in:
parent
bcf4fd6bd3
commit
33503ea248
4 changed files with 4 additions and 69 deletions
|
|
@ -215,21 +215,6 @@ TEMPLATE_TO_MODEL_MAPPER = {
|
|||
"google/gemma-3n-E2B-it",
|
||||
"unsloth/gemma-3n-E2B-it-unsloth-bnb-4bit",
|
||||
),
|
||||
"gemma-4": (
|
||||
"unsloth/gemma-4-E2B-it",
|
||||
"google/gemma-4-E2B-it",
|
||||
"unsloth/gemma-4-E4B-it",
|
||||
"google/gemma-4-E4B-it",
|
||||
"unsloth/gemma-4-E2B-it-unsloth-bnb-4bit",
|
||||
"unsloth/gemma-4-E4B-it-unsloth-bnb-4bit",
|
||||
),
|
||||
"gemma-4-thinking": (
|
||||
"unsloth/gemma-4-26B-A4B-it",
|
||||
"google/gemma-4-26B-A4B-it",
|
||||
"unsloth/gemma-4-31B-it",
|
||||
"unsloth/gemma-4-31B-it-unsloth-bnb-4bit",
|
||||
"google/gemma-4-31B-it",
|
||||
),
|
||||
"qwen2.5": (
|
||||
"unsloth/Qwen2.5-0.5B-Instruct-unsloth-bnb-4bit",
|
||||
"unsloth/Qwen2.5-0.5B-Instruct",
|
||||
|
|
@ -414,12 +399,6 @@ TEMPLATE_TO_MODEL_MAPPER = {
|
|||
"THUDM/GLM-4.7-Flash",
|
||||
"unsloth/GLM-4.7-Flash-bnb-4bit",
|
||||
),
|
||||
"lfm-2": (
|
||||
"unsloth/LFM2-1.2B",
|
||||
"LiquidAI/LFM2-1.2B",
|
||||
"unsloth/LFM2-1.2B-unsloth-bnb-4bit",
|
||||
),
|
||||
"lfm-2.5": ("unsloth/LFM2.5-1.2B-Instruct",),
|
||||
}
|
||||
|
||||
MODEL_TO_TEMPLATE_MAPPER = {}
|
||||
|
|
@ -435,14 +414,6 @@ for key, values in TEMPLATE_TO_MODEL_MAPPER.items():
|
|||
|
||||
|
||||
TEMPLATE_TO_RESPONSES_MAPPER = {
|
||||
"gemma-4-thinking": {
|
||||
"instruction": "<|turn>user\n",
|
||||
"response": "<|turn>model\n",
|
||||
},
|
||||
"gemma-4": {
|
||||
"instruction": "<|turn>user\n",
|
||||
"response": "<|turn>model\n",
|
||||
},
|
||||
"gemma-3": {
|
||||
"instruction": "<start_of_turn>user\n",
|
||||
"response": "<start_of_turn>model\n",
|
||||
|
|
@ -543,10 +514,6 @@ TEMPLATE_TO_RESPONSES_MAPPER = {
|
|||
"instruction": "<|im_start|>user\n",
|
||||
"response": "<|im_start|>assistant\n",
|
||||
},
|
||||
"lfm-2.5": {
|
||||
"instruction": "<|im_start|>user\n",
|
||||
"response": "<|im_start|>assistant\n",
|
||||
},
|
||||
"starling": {
|
||||
"instruction": "GPT4 Correct User: ",
|
||||
"response": "GPT4 Correct Assistant: ",
|
||||
|
|
|
|||
|
|
@ -52,7 +52,6 @@ TRANSFORMERS_5_MODEL_SUBSTRINGS: tuple[str, ...] = (
|
|||
"qwen3.5", # Qwen3.5 family (35B-A3B, etc.)
|
||||
"qwen3-next", # Qwen3-Next and variants
|
||||
"tiny_qwen3_moe", # imdatta0/tiny_qwen3_moe_2.8B_0.7B
|
||||
"lfm2.5-vl-450m", # LiquidAI/LFM2.5-VL-450M
|
||||
)
|
||||
|
||||
# Lowercase substrings for models that require transformers 5.5.0 (checked first).
|
||||
|
|
|
|||
|
|
@ -22,39 +22,6 @@ __all__ = [
|
|||
|
||||
__INT_TO_FLOAT_MAPPER = \
|
||||
{
|
||||
"unsloth/gemma-4-E2B-it-unsloth-bnb-4bit" : (
|
||||
"unsloth/gemma-4-E2B-it",
|
||||
"google/gemma-4-E2B-it",
|
||||
),
|
||||
"unsloth/gemma-4-E4B-it-unsloth-bnb-4bit" : (
|
||||
"unsloth/gemma-4-E4B-it",
|
||||
"google/gemma-4-E4B-it",
|
||||
),
|
||||
"unsloth/gemma-4-26B-A4B-it" : (
|
||||
"unsloth/gemma-4-26B-A4B-it",
|
||||
"google/gemma-4-26B-A4B-it",
|
||||
),
|
||||
"unsloth/gemma-4-31B-it" : (
|
||||
"unsloth/gemma-4-31B-it",
|
||||
"google/gemma-4-31B-it",
|
||||
),
|
||||
"unsloth/gemma-4-E2B" : (
|
||||
"unsloth/gemma-4-E2B",
|
||||
"google/gemma-4-E2B",
|
||||
),
|
||||
"unsloth/gemma-4-E4B" : (
|
||||
"unsloth/gemma-4-E4B",
|
||||
"google/gemma-4-E4B",
|
||||
),
|
||||
"unsloth/gemma-4-26B-A4B" : (
|
||||
"unsloth/gemma-4-26B-A4B",
|
||||
"google/gemma-4-26B-A4B",
|
||||
),
|
||||
"unsloth/gemma-4-31B" : (
|
||||
"unsloth/gemma-4-31B",
|
||||
"google/gemma-4-31B",
|
||||
),
|
||||
|
||||
"unsloth/mistral-7b-bnb-4bit" : (
|
||||
"unsloth/mistral-7b",
|
||||
"mistralai/Mistral-7B-v0.1",
|
||||
|
|
|
|||
|
|
@ -1978,11 +1978,13 @@ OLLAMA_TEMPLATE_TO_MODEL_MAPPER = {
|
|||
),
|
||||
"gemma4": (
|
||||
"unsloth/gemma-4-E2B-it",
|
||||
"unsloth/gemma-4-E2B-it-unsloth-bnb-4bit",
|
||||
"unsloth/gemma-4-E2B",
|
||||
"unsloth/gemma-4-E4B-it",
|
||||
"unsloth/gemma-4-E4B-it-unsloth-bnb-4bit",
|
||||
"unsloth/gemma-4-E4B",
|
||||
"unsloth/gemma-4-31B-it",
|
||||
"unsloth/gemma-4-31B",
|
||||
"unsloth/gemma-4-26B-A4B-it",
|
||||
"unsloth/gemma-4-26B-A4B",
|
||||
),
|
||||
"gemma3n": (
|
||||
"unsloth/gemma-3n-E4B-it-unsloth-bnb-4bit",
|
||||
|
|
|
|||
Loading…
Reference in a new issue