Skip to content

Commit 65afe7b

Browse files
nits
1 parent 830017c commit 65afe7b

File tree

1 file changed

+0
-1
lines changed

1 file changed

+0
-1
lines changed

mlx_vlm/trainer/trainer.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -68,7 +68,6 @@ class TrainingArgs:
6868
def default_loss(model, inputs, targets, lengths, train_on_completions=False, assistant_id=77091):
6969
outputs = model(inputs)
7070
logits = outputs.logits.astype(mx.float32)
71-
7271
_, seq_len = targets.shape
7372
steps = mx.arange(seq_len)[None, :]
7473
base_mask = steps < lengths[:, None]

0 commit comments

Comments
 (0)