From 0cb5ec84de70d3bd80ecad06b2e91125784dac7e Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Tue, 17 Sep 2024 10:50:57 -0700 Subject: [PATCH] Update mapper.py --- unsloth/models/mapper.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/unsloth/models/mapper.py b/unsloth/models/mapper.py index 08a9ef2097..615480c678 100644 --- a/unsloth/models/mapper.py +++ b/unsloth/models/mapper.py @@ -300,6 +300,10 @@ __INT_TO_FLOAT_MAPPER = \ "unsloth/SmolLM-1.7B-Instruct", "HuggingFaceTB/SmolLM-1.7B-Instruct", ), + "unsloth/Mistral-Small-Instruct-2409-bnb-4bit" : ( + "unsloth/Mistral-Small-Instruct-2409", + "mistralai/Mistral-Small-Instruct-2409", + ), } INT_TO_FLOAT_MAPPER = {}