F.cross_entropy(logits, targets, ignore_index=NUM_AUDIO_TOKENS)