From bb45fdabb6643ba2b85b3530f2c07daa681d936b Mon Sep 17 00:00:00 2001 From: Daniel Han-Chen Date: Sun, 24 Mar 2024 14:00:04 +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 afcbdb75f5..dc54bfd791 100644 --- a/unsloth/models/mapper.py +++ b/unsloth/models/mapper.py @@ -90,6 +90,10 @@ __INT_TO_FLOAT_MAPPER = \ "unsloth/gemma-2b-it", "google/gemma-2b-it", ), + "unsloth/mistral-7b-v0.2-bnb-4bit" : ( + "unsloth/mistral-7b-v0.2", + "alpindale/Mistral-7B-v0.2-hf", + ) } INT_TO_FLOAT_MAPPER = {}