tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

fix input args

Browse Source
This commit is contained in:
Wing Lian
2025-05-20 07:34:41 -07:00
parent 6fafe46562
commit 83ad248e5b
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/axolotl/integrations/kd/trainer.py
View File

@@ -92,7 +92,7 @@ class AxolotlKDTrainer(AxolotlTrainer):
) )
else: else:
loss_kd = self.kd_loss_fn( loss_kd = self.kd_loss_fn(
# shift_logits, shift_logits,
target_token_ids_for_loss, target_token_ids_for_loss,
target_logprobs_for_loss, target_logprobs_for_loss,
target_mask_for_loss, target_mask_for_loss,