don't need temp arg to distill method

2025-05-23 10:42:25 -04:00
parent 49e2fa825d
commit a8d9fab635
1 changed files with 0 additions and 1 deletions
--- a/src/axolotl/integrations/kd/kernels/liger.py
+++ b/src/axolotl/integrations/kd/kernels/liger.py
@@ -125,7 +125,6 @@ class LigerFusedLinearKLTopKLogprobFunction(LigerFusedLinearDistillationBase):
                target_token_ids_chunk,
                target_logprobs_chunk,
                target_mask_chunk,
-                temperature=temperature,
            )

        loss = weight_hard_loss * ce_loss + weight_soft_loss * soft_loss