Update loader.py

2024-11-21 02:18:59 -08:00 · 2024-11-21 02:18:59 -08:00 · 646c1ddb67
commit 646c1ddb67
parent 0d017bc6cf
1 changed files with 1 additions and 0 deletions
--- 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,
        )