From 646c1ddb677f32947695d62f3cd5571e013789a4 Mon Sep 17 00:00:00 2001
From: Daniel Han <danielhanchen@gmail.com>
Date: Thu, 21 Nov 2024 02:18:59 -0800
Subject: [PATCH] Update loader.py

---
 unsloth/models/loader.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/unsloth/models/loader.py b/unsloth/models/loader.py
index 6eb8ed7863..6d388900de 100644
--- a/unsloth/models/loader.py
+++ b/unsloth/models/loader.py
@@ -490,6 +490,7 @@ class FastVisionModel(FastBaseVisionModel):
             device_map        = device_map,
             trust_remote_code = trust_remote_code,
             revision          = revision if not is_peft else None,
+            model_types       = model_types,
             *args, **kwargs,
         )