From 41f1783505cb0112bc492bf9a83e8f115c50802d Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Sat, 15 Feb 2025 02:48:36 -0800 Subject: [PATCH] Update mapper.py --- unsloth/models/mapper.py | 31 +++++++++++++++++++++++++++++-- 1 file changed, 29 insertions(+), 2 deletions(-) diff --git a/unsloth/models/mapper.py b/unsloth/models/mapper.py index c81290b662..2e85d30145 100644 --- a/unsloth/models/mapper.py +++ b/unsloth/models/mapper.py @@ -218,13 +218,25 @@ __INT_TO_FLOAT_MAPPER = \ "unsloth/Mistral-Nemo-Base-2407", "mistralai/Mistral-Nemo-Base-2407", ), - "unsloth/Meta-Llama-3.1-8B-bnb-4bit" : ( + "unsloth/Meta-Llama-3.1-8B-unsloth-bnb-4bit" : ( "unsloth/Meta-Llama-3.1-8B", "meta-llama/Meta-Llama-3.1-8B", + "unsloth/Meta-Llama-3.1-8B-bnb-4bit", ), - "unsloth/Meta-Llama-3.1-8B-Instruct-bnb-4bit" : ( + "unsloth/Meta-Llama-3.1-8B-Instruct-unsloth-bnb-4bit" : ( "unsloth/Meta-Llama-3.1-8B-Instruct", "meta-llama/Meta-Llama-3.1-8B-Instruct", + "unsloth/Meta-Llama-3.1-8B-Instruct-bnb-4bit", + ), + "unsloth/Llama-3.1-8B-unsloth-bnb-4bit" : ( + "unsloth/Llama-3.1-8B", + "meta-llama/Llama-3.1-8B", + "unsloth/Llama-3.1-8B-bnb-4bit", + ), + "unsloth/Llama-3.1-8B-Instruct-unsloth-bnb-4bit" : ( + "unsloth/Llama-3.1-8B-Instruct", + "meta-llama/Llama-3.1-8B-Instruct", + "unsloth/Llama-3.1-8B-Instruct-bnb-4bit", ), "unsloth/Meta-Llama-3.1-70B-bnb-4bit" : ( "unsloth/Meta-Llama-3.1-70B", @@ -589,6 +601,21 @@ __INT_TO_FLOAT_MAPPER = \ "Qwen/Qwen2.5-VL-72B-Instruct", "unsloth/Qwen2.5-VL-72B-Instruct-bnb-4bit", ), + "unsloth/DeepHermes-3-Llama-3-8B-Preview-unsloth-bnb-4bit" : ( + "unsloth/DeepHermes-3-Llama-3-8B-Preview", + "NousResearch/DeepHermes-3-Llama-3-8B-Preview", + "unsloth/DeepHermes-3-Llama-3-8B-Preview-bnb-4bit", + ), + "unsloth/DeepScaleR-1.5B-Preview-unsloth-bnb-4bit" : ( + "unsloth/DeepHermes-3-Llama-3-8B-Preview", + "agentica-org/DeepScaleR-1.5B-Preview", + "unsloth/DeepScaleR-1.5B-Preview-bnb-4bit", + ), + "unsloth/OpenThinker-7B-unsloth-bnb-4bit" : ( + "unsloth/OpenThinker-7B", + "open-thoughts/OpenThinker-7B", + "unsloth/OpenThinker-7B-bnb-4bit", + ), } INT_TO_FLOAT_MAPPER = {}