From 84d3e6f1a91224590fd37e98902bc057cf6162c4 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Thu, 26 Jun 2025 03:55:07 -0700 Subject: [PATCH] Update vision.py --- unsloth/models/vision.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/unsloth/models/vision.py b/unsloth/models/vision.py index b0f485d90e..8dca483783 100644 --- a/unsloth/models/vision.py +++ b/unsloth/models/vision.py @@ -722,6 +722,8 @@ class FastBaseModel: pass # Must disable returning hidden states in the case for GRPO os.environ["UNSLOTH_RETURN_HIDDEN_STATES"] = "0" + # Must enable returning logits + os.environ["UNSLOTH_RETURN_LOGITS"] = "1" return model pass @@ -760,6 +762,8 @@ class FastBaseModel: embeddings = model.get_output_embeddings() if hasattr(embeddings, "training"): embeddings.training = True pass + # Can re-enable not returning logits + os.environ["UNSLOTH_RETURN_LOGITS"] = "0" return model pass pass