From d32ee870a70a4bb3ade8658c76f64ed9698bc0a8 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 08:32:56 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- unsloth/models/rl_replacements.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/unsloth/models/rl_replacements.py b/unsloth/models/rl_replacements.py index 8d7fa91b74..faba17339b 100755 --- a/unsloth/models/rl_replacements.py +++ b/unsloth/models/rl_replacements.py @@ -865,7 +865,9 @@ def grpo_trainer__get_per_token_logps_and_entropies(function_name, function): if token_type_ids_chunk is not None: _extra_vision_kwargs["token_type_ids"] = token_type_ids_chunk if mm_token_type_ids_chunk is not None: - _extra_vision_kwargs["mm_token_type_ids"] = mm_token_type_ids_chunk + _extra_vision_kwargs["mm_token_type_ids"] = ( + mm_token_type_ids_chunk + ) with torch.amp.autocast( device_type = "cuda", dtype = self._autocast_dtype ):