From 3effbaf319c27d7eadab4fa3aac84991d78e7a1c Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Mon, 20 Jan 2025 08:10:20 -0800 Subject: [PATCH] Update mapper.py --- unsloth/models/mapper.py | 28 ++++++++++++++++++++++++++++ 1 file changed, 28 insertions(+) diff --git a/unsloth/models/mapper.py b/unsloth/models/mapper.py index b7df6668bb..72619cf05d 100644 --- a/unsloth/models/mapper.py +++ b/unsloth/models/mapper.py @@ -522,6 +522,34 @@ __INT_TO_FLOAT_MAPPER = \ "microsoft/phi-4", "unsloth/phi-4-bnb-4bit", ), + "unsloth/DeepSeek-R1-Distill-Qwen-32B-bnb-4bit" : ( + "unsloth/DeepSeek-R1-Distill-Qwen-32B", + "deepseek-ai/DeepSeek-R1-Distill-Qwen-32B", + ), + "unsloth/DeepSeek-R1-Distill-Qwen-14B-unsloth-bnb-4bit" : ( + "unsloth/DeepSeek-R1-Distill-Qwen-14B", + "deepseek-ai/DeepSeek-R1-Distill-Qwen-14B", + "unsloth/DeepSeek-R1-Distill-Qwen-14B-bnb-4bit", + ), + "unsloth/DeepSeek-R1-Distill-Qwen-7B-unsloth-bnb-4bit" : ( + "unsloth/DeepSeek-R1-Distill-Qwen-7B", + "deepseek-ai/DeepSeek-R1-Distill-Qwen-7B", + "unsloth/DeepSeek-R1-Distill-Qwen-7B-bnb-4bit", + ), + "unsloth/DeepSeek-R1-Distill-Qwen-1.5B-unsloth-bnb-4bit" : ( + "unsloth/DeepSeek-R1-Distill-Qwen-1.5B", + "deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B", + "unsloth/DeepSeek-R1-Distill-Qwen-1.5B-bnb-4bit", + ), + "unsloth/DeepSeek-R1-Distill-Llama-8B-unsloth-bnb-4bit" : ( + "unsloth/DeepSeek-R1-Distill-Llama-8B", + "deepseek-ai/DeepSeek-R1-Distill-Llama-8B", + "unsloth/DeepSeek-R1-Distill-Llama-8B-bnb-4bit", + ), + "unsloth/DeepSeek-R1-Distill-Llama-70B-bnb-4bit" : ( + "unsloth/DeepSeek-R1-Distill-Llama-70B", + "deepseek-ai/DeepSeek-R1-Distill-Llama-70B", + ), } INT_TO_FLOAT_MAPPER = {}