Method set_conv_state

cake-core/src/models/common/cache.rs:236–238

Set the conv state for a linear attention layer.

(&mut self, block_idx: usize, state: Tensor)

    /// Set the conv state for a linear attention layer.
    pub fn set_conv_state(&mut self, block_idx: usize, state: Tensor) {
        self.conv_states[block_idx] = Some(state);
    }

    /// Return a copy of this cache with the same state but new kv table.
    pub fn as_new(&self) -> Self {