Method take_slot

cake-core/src/models/vibevoice/vae_decoder.rs:36–44

Take the next cache slot, returning (index, is_first_use).

(&mut self)

    /// Take the next cache slot, returning (index, is_first_use).
    pub fn take_slot(&mut self) -> (usize, bool) {
        let idx = self.counter;
        self.counter += 1;
        if idx >= self.states.len() {
            self.states.resize(idx + 1, None);
        }
        let is_first = self.states[idx].is_none();
        (idx, is_first)
    }

    pub fn get(&self, idx: usize) -> Option<&Tensor> {
        self.states.get(idx).and_then(|s| s.as_ref())