From 0c932bc0bb79b405af6e4b623088c86bdc51e48e Mon Sep 17 00:00:00 2001
From: Daniel Han <danielhanchen@gmail.com>
Date: Tue, 30 Jul 2024 23:51:47 -0700
Subject: [PATCH] Update mapper.py

---
 unsloth/models/mapper.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/unsloth/models/mapper.py b/unsloth/models/mapper.py
index 462555f317..57ba676585 100644
--- a/unsloth/models/mapper.py
+++ b/unsloth/models/mapper.py
@@ -241,6 +241,14 @@ __INT_TO_FLOAT_MAPPER = \
     "unsloth/Mistral-Large-Instruct-2407-bnb-4bit" : (
         "mistralai/Mistral-Large-Instruct-2407",
     ),
+    "unsloth/gemma-2-2b-bnb-4bit" : (
+        "unsloth/gemma-2-2b",
+        "google/gemma-2-2b",
+    ),
+    "unsloth/gemma-2-2b-it-bnb-4bit" : (
+        "unsloth/gemma-2-2b-it",
+        "google/gemma-2-2b-it",
+    ),
 }
 
 INT_TO_FLOAT_MAPPER = {}