def __init__()

in optimum/executorch/attentions/custom_kv_cache.py [0:0]

• 8 lines of code
• 1 McCabe index (conditional complexity)

    def __init__(
        self,
        config,
        max_batch_size: int,
        max_cache_len: Optional[int] = None,
        device: Union[torch.device, str, None] = None,
        dtype: torch.dtype = torch.float32,
        layer_device_map: Optional[Dict[int, Union[str, torch.device, int]]] = None,