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

fix trainer callback base class

Browse Source
This commit is contained in:
Wing Lian
2025-05-28 00:46:52 -04:00
parent 20602fd93f
commit 3a0faa97ca
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/axolotl/integrations/kd/callbacks.py
View File

@@ -4,10 +4,10 @@ Transformers trainer callbacks to schedule the KD temperature during training
import math import math
from transformers.trainer_callback import CallbackHandler from transformers.trainer_callback import TrainerCallback
class KDTemperatureSchedulerCallback(CallbackHandler): class KDTemperatureSchedulerCallback(TrainerCallback):
""" """
KD temperature scheduler callback for the trainer. KD temperature scheduler callback for the trainer.
""" """