Update mapper.py

unslothai · Jul 31, 2024 · 0c932bc · 0c932bc
1 parent 8db7e80
commit 0c932bc
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/unsloth/models/mapper.py b/unsloth/models/mapper.py
@@ -241,6 +241,14 @@
     "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 = {}