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

disable _is_vlm (#3265)

Browse source
This commit is contained in:
DoubleMathew 2025-09-04 06:52:35 -05:00 • committed by GitHub
commit cd71b1bb96
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 7 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

7
unsloth/models/rl.py
View file

@ -727,6 +727,13 @@ def _patch_trl_rl_trainers(trainer_file = "grpo_trainer"):
new_text = 'self._signature_columns = ["input_ids", "attention_mask", "completion_mask","labels"]'
RLTrainer_source = RLTrainer_source.replace(original_text, new_text)
# Temporary patch _is_vlm to False
# as of 0.22 it only exists in sfttrainer
oriignal_is_vlm_text = 'self._is_vlm = True'
new_is_vlm_text = 'self._is_vlm = False'
RLTrainer_source = RLTrainer_source.replace(oriignal_is_vlm_text, new_is_vlm_text)
# Remove multiple doc strings
if __RLConfig_doc__ != "" and RLTrainer_source.count(__RLTrainer_doc__) == 2:
RLTrainer_source = RLTrainer_source.replace(__RLTrainer_doc__, "", 1)