From 84b37f8548912764345fbc3db7ede9099b92d916 Mon Sep 17 00:00:00 2001 From: Daniel Han-Chen Date: Thu, 15 Feb 2024 03:53:58 +1100 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 1c90a459b8..323358fff1 100644 --- a/unsloth/models/mapper.py +++ b/unsloth/models/mapper.py @@ -42,6 +42,10 @@ __INT_TO_FLOAT_MAPPER = \ "unsloth/tinyllama", "TinyLlama/TinyLlama-1.1B-intermediate-step-1431k-3T", ), + "unsloth/tinyllama-chat-bnb-4bit" : ( + "unsloth/tinyllama-chat", + "TinyLlama/TinyLlama-1.1B-Chat-v1.0", + ), "unsloth/mistral-7b-instruct-v0.1-bnb-4bit" : ( "mistralai/Mistral-7B-Instruct-v0.1", ),