lambda-160m / kv_cache.py
MK0727's picture
Upload lambda-160m pretrained model
134df9b verified
raw
history blame contribute delete
110 Bytes
import torch
LayerKeyValueCache = tuple[torch.Tensor, torch.Tensor]
KeyValueCache = list[LayerKeyValueCache]