vllm.model_executor.layers.mamba.kda_checkpoint

Classes:

• FlashKDAPrefillCheckpointExporter –

  Store FlashKDA recurrent and convolution checkpoint states.

FlashKDAPrefillCheckpointExporter dataclass

Bases: MambaPrefillCheckpointExporter

Store FlashKDA recurrent and convolution checkpoint states.

Source code in vllm/model_executor/layers/mamba/kda_checkpoint.py

class FlashKDAPrefillCheckpointExporter(MambaPrefillCheckpointExporter):
    """Store FlashKDA recurrent and convolution checkpoint states."""