ros/LlamaFactory
1
0
Fork 0
mirror of https://github.com/hiyouga/LlamaFactory.git synced 2026-02-03 21:03:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

[trainer] fix vlm loss for transformers 4.49 (#7448)

Browse Source
This commit is contained in:
hoshi-hiyouga
2025-03-24 10:24:05 +08:00
committed by GitHub
parent 3612946dd9
commit 7203365b80
5 changed files with 21 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/llamafactory/train/kto/trainer.py
View File

@@ -127,9 +127,9 @@ class CustomKTOTrainer(KTOTrainer):
return Trainer._get_train_sampler(self)
@override
def get_batch_samples(self, epoch_iterator, num_batches, *args, **kwargs):
def get_batch_samples(self, *args, **kwargs):
r"""Replace the method of KTO Trainer with the one of the standard Trainer."""
return Trainer.get_batch_samples(self, epoch_iterator, num_batches, *args, **kwargs)
return Trainer.get_batch_samples(self, *args, **kwargs)
@override
def forward(