IMPORTANT: To view this page as Markdown, append `.md` to the URL (e.g. /get-started.md). For the complete documentation index, see llms.txt.
Skip to main content
For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /get-started.md).

Mojo trait

KVCollectionT

Trait for a pair of caches (keys and values).

Implemented traits​

AnyType, Copyable, ImplicitlyCopyable, Movable

comptime members​

CacheType​

comptime CacheType

dtype​

comptime dtype

kv_params​

comptime kv_params

name_str​

comptime name_str

Required methods​

get_key_cache​

def get_key_cache(self, layer_idx: Int) -> Self.CacheType

Returns:

_Self.CacheType

get_value_cache​

def get_value_cache(self, layer_idx: Int) -> Self.CacheType

Returns:

_Self.CacheType

cache_length​

def cache_length(self, bs_idx: Int) -> Int

Returns:

Int

Was this page helpful?