From 17bbba3dad28c46d7c0a323973bfee937e860c29 Mon Sep 17 00:00:00 2001
From: Daniel Han <danielhanchen@gmail.com>
Date: Wed, 20 Nov 2024 04:15:53 -0800
Subject: [PATCH] vision

---
 unsloth/models/loader.py |  5 +++--
 unsloth/models/mapper.py | 32 ++++++++++++++++++++++++++++++++
 2 files changed, 35 insertions(+), 2 deletions(-)

diff --git a/unsloth/models/loader.py b/unsloth/models/loader.py
index 7a6322d248..1d4bb71e85 100644
--- a/unsloth/models/loader.py
+++ b/unsloth/models/loader.py
@@ -153,8 +153,9 @@ def get_model_name(model_name, load_in_4bit = True):
         if upgraded_model_name is not None:
             raise NotImplementedError(
                 f"Unsloth: {model_name} is not supported in your current Unsloth version! Please update Unsloth via:\n\n"\
-                'pip uninstall unsloth -y\n'\
-                'pip install --upgrade --no-cache-dir "unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git"'
+                'pip uninstall unsloth unsloth_zoo -y\n'\
+                'pip install --upgrade --no-cache-dir "unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git"\n'\
+                'pip install --upgrade --no-cache-dir "git+https://github.com/unslothai/unsloth-zoo.git"\n'\
             )
         pass
     pass
diff --git a/unsloth/models/mapper.py b/unsloth/models/mapper.py
index 627aab2054..03047b8103 100644
--- a/unsloth/models/mapper.py
+++ b/unsloth/models/mapper.py
@@ -452,6 +452,38 @@ __INT_TO_FLOAT_MAPPER = \
         "unsloth/Llama-3.1-Nemotron-70B-Instruct",
         "nvidia/Llama-3.1-Nemotron-70B-Instruct-HF",
     ),
+    "unsloth/Qwen2-VL-2B-Instruct-bnb-4bit" : (
+        "unsloth/Qwen2-VL-2B-Instruct",
+        "Qwen/Qwen2-VL-2B-Instruct",
+    ),
+    "unsloth/Qwen2-VL-7B-Instruct-bnb-4bit" : (
+        "unsloth/Qwen2-VL-7B-Instruct",
+        "Qwen/Qwen2-VL-7B-Instruct",
+    ),
+    "unsloth/Llama-3.2-11B-Vision-Instruct-bnb-4bit" : (
+        "unsloth/Llama-3.2-11B-Vision-Instruct",
+        "meta-llama/Llama-3.2-11B-Vision-Instruct",
+    ),
+    "unsloth/Llama-3.2-90B-Vision-Instruct-bnb-4bit" : (
+        "unsloth/Llama-3.2-90B-Vision-Instruct",
+        "meta-llama/Llama-3.2-90B-Vision-Instruct",
+    ),
+    "unsloth/Llama-3.2-11B-Vision-bnb-4bit" : (
+        "unsloth/Llama-3.2-11B-Vision",
+        "meta-llama/Llama-3.2-11B-Vision",
+    ),
+    "unsloth/Llama-3.2-90B-Vision-bnb-4bit" : (
+        "unsloth/Llama-3.2-90B-Vision",
+        "meta-llama/Llama-3.2-90B-Vision",
+    ),
+    "unsloth/Pixtral-12B-2409-bnb-4bit" : (
+        "unsloth/Pixtral-12B-2409",
+        "mistralai/Pixtral-12B-2409",
+    ),
+    "unsloth/Pixtral-12B-2409-Base-bnb-4bit" : (
+        "unsloth/Pixtral-12B-Base-2409",
+        "mistralai/Pixtral-12B-Base-2409",
+    ),
 }
 
 INT_TO_FLOAT_MAPPER  = {}