Remove chunk arg

Signed-off-by: Austin Liu <[email protected]>
linkedin · Dec 4, 2024 · 5257d26 · 5257d26
1 parent 805e50f
commit 5257d26
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/src/liger_kernel/chunked_loss/fused_linear_distillation.py b/src/liger_kernel/chunked_loss/fused_linear_distillation.py
@@ -103,7 +103,6 @@ def forward(
             LigerFusedLinearDistillationBase._compute_loss,
             distillation_loss_fn=loss_fn,
             full_target=target,
-            chunk_size=chunk_size,
             ignore_index=ignore_index,
             weight_hard_loss=weight_hard_loss,
             weight_soft_loss=weight_soft_loss,
@@ -202,7 +201,6 @@ def _compute_loss(
         teacher_bias=None,
         distillation_loss_fn=None,
         full_target=None,
-        chunk_size=1,
         ignore_index=-100,
         temperature=1.0,
         weight_hard_loss=0.5,