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]