ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

[pre-commit.ci] auto fixes from pre-commit.com hooks

Browse source 
for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2026-04-01 08:32:56 +00:00
commit d32ee870a7
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
unsloth/models/rl_replacements.py
View file

@ -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
):