From b506fbd86f82e4fd09fb729235315c5b2f9d3188 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Wed, 1 Apr 2026 07:28:55 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- unsloth/models/loader.py | 1 + 1 file changed, 1 insertion(+) diff --git a/unsloth/models/loader.py b/unsloth/models/loader.py index 951fe6d228..9b7b1b02c5 100644 --- a/unsloth/models/loader.py +++ b/unsloth/models/loader.py @@ -1412,6 +1412,7 @@ class FastModel(FastBaseModel): if auto_model is None: if _num_labels is not None: from transformers import AutoModelForSequenceClassification + auto_model = AutoModelForSequenceClassification elif is_vlm: # Check if the model's auto_map supports the VLM auto class.