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", ),