Function candle_transformers::utils::apply_repeat_penalty

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