enable slots=True for dataclasses

2026-02-08 23:46:01 -05:00
parent 2f21442653
commit f438ce463f
3 changed files with 3 additions and 3 deletions
@@ -2,7 +2,7 @@ from dataclasses import dataclass
 import torch


-@dataclass
+@dataclass(slots=True)
 class Context:
    is_prefill: bool = False
    cu_seqlens_q: torch.Tensor | None = None