Skip to main content

Python class

ArchConfigWithKVCache

ArchConfigWithKVCache

class max.pipelines.lib.interfaces.ArchConfigWithKVCache(*args, **kwargs)

source

Bases: ArchConfig, Protocol

Config for a model architecture that uses a KV cache.

get_kv_params()

get_kv_params()

source

KV cache parameters to use when running the model.

Return type:

KVCacheParamInterface