From d6fd96cc86173a928f6d87a83cf83470df18abb9 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Sun, 9 Mar 2025 23:13:30 -0700 Subject: [PATCH] Update vision.py --- unsloth/models/vision.py | 5 ----- 1 file changed, 5 deletions(-) diff --git a/unsloth/models/vision.py b/unsloth/models/vision.py index c8cad90152..2830783756 100644 --- a/unsloth/models/vision.py +++ b/unsloth/models/vision.py @@ -366,11 +366,6 @@ class FastBaseModel: model, use_gradient_checkpointing = True, ): - if not isinstance(model, PeftModelForCausalLM): - raise TypeError( - "Unsloth: Your model needs to call `.get_peft_model` first!" - ) - pass full_finetuning = os.environ.get("UNSLOTH_ENABLE_FULL_FINETUNING", "0") == "1" float32_mixed_precision = True