Function apply_repeat_penalty 

pub fn apply_repeat_penalty(
    logits: &Tensor,
    penalty: f32,
    context: &[u32],
) -> Result<Tensor>