Struct PageGuard 

pub struct PageGuard {
    pub expected: usize,
    pub page_id: u32,
    pub eos_id: u32,
    pub max_page_tokens: usize,
    /* private fields */
}

_uocr_page_count_guard: in place on the logits, over the generated ids.

Fields

expected: usize

page_id: u32

eos_id: u32

max_page_tokens: usize

Implementations

impl PageGuard

pub fn new(expected: usize, page_id: u32, eos_id: u32) -> Self

pub fn apply(&mut self, logits: &mut [f32], generated: &[u32])