From bd2fa264c9027367cb5b5458d03cb6d623f12e9e Mon Sep 17 00:00:00 2001
From: Daniel Han-Chen <danielhanchen@gmail.com>
Date: Fri, 23 Feb 2024 01:57:34 +1100
Subject: [PATCH] Update mapper.py

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

diff --git a/unsloth/models/mapper.py b/unsloth/models/mapper.py
index 323358fff1..afcbdb75f5 100644
--- a/unsloth/models/mapper.py
+++ b/unsloth/models/mapper.py
@@ -74,6 +74,22 @@ __INT_TO_FLOAT_MAPPER = \
     "unsloth/solar-10.7b-bnb-4bit" : (
         "upstage/SOLAR-10.7B-v1.0",
     ),
+    "unsloth/gemma-7b-bnb-4bit" : (
+        "unsloth/gemma-7b",
+        "google/gemma-7b",
+    ),
+    "unsloth/gemma-2b-bnb-4bit" : (
+        "unsloth/gemma-2b",
+        "google/gemma-2b",
+    ),
+    "unsloth/gemma-7b-it-bnb-4bit" : (
+        "unsloth/gemma-7b-it",
+        "google/gemma-7b-it",
+    ),
+    "unsloth/gemma-2b-bnb-4bit" : (
+        "unsloth/gemma-2b-it",
+        "google/gemma-2b-it",
+    ),
 }
 
 INT_TO_FLOAT_MAPPER = {}