We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 830017c commit 65afe7bCopy full SHA for 65afe7b
mlx_vlm/trainer/trainer.py
@@ -68,7 +68,6 @@ class TrainingArgs:
68
def default_loss(model, inputs, targets, lengths, train_on_completions=False, assistant_id=77091):
69
outputs = model(inputs)
70
logits = outputs.logits.astype(mx.float32)
71
-
72
_, seq_len = targets.shape
73
steps = mx.arange(seq_len)[None, :]
74
base_mask = steps < lengths[:, None]
0 commit comments