From 62c989ef0ae0e9fbac714a4cb21eda76c1fe84b6 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Sun, 15 Sep 2024 21:50:00 -0700 Subject: [PATCH] Update mapper.py --- unsloth/models/mapper.py | 24 ++++++++++++++++++++++++ 1 file changed, 24 insertions(+) diff --git a/unsloth/models/mapper.py b/unsloth/models/mapper.py index bff7f025..08a9ef20 100644 --- a/unsloth/models/mapper.py +++ b/unsloth/models/mapper.py @@ -276,6 +276,30 @@ "unsloth/Hermes-3-Llama-3.1-405B-bnb-4bit" : ( "NousResearch/Hermes-3-Llama-3.1-405B", ), + "unsloth/SmolLM-135M-bnb-4bit" : ( + "unsloth/SmolLM-135M", + "HuggingFaceTB/SmolLM-135M", + ), + "unsloth/SmolLM-360M-bnb-4bit" : ( + "unsloth/SmolLM-360M", + "HuggingFaceTB/SmolLM-360M", + ), + "unsloth/SmolLM-1.7B-bnb-4bit" : ( + "unsloth/SmolLM-1.7B", + "HuggingFaceTB/SmolLM-1.7B", + ), + "unsloth/SmolLM-135M-Instruct-bnb-4bit" : ( + "unsloth/SmolLM-135M-Instruct", + "HuggingFaceTB/SmolLM-135M-Instruct", + ), + "unsloth/SmolLM-360M-Instruct-bnb-4bit" : ( + "unsloth/SmolLM-360M-Instruct", + "HuggingFaceTB/SmolLM-360M-Instruct", + ), + "unsloth/SmolLM-1.7B-Instruct-bnb-4bit" : ( + "unsloth/SmolLM-1.7B-Instruct", + "HuggingFaceTB/SmolLM-1.7B-Instruct", + ), } INT_TO_FLOAT_MAPPER = {}