From f438ce463f24700fb1d4671934abd2714d9e865f Mon Sep 17 00:00:00 2001 From: IceCreamMilkyTea Date: Sun, 8 Feb 2026 23:46:01 -0500 Subject: [PATCH] enable slots=True for dataclasses --- nanovllm/config.py | 2 +- nanovllm/sampling_params.py | 2 +- nanovllm/utils/context.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/nanovllm/config.py b/nanovllm/config.py index 959ffb3..b999838 100644 --- a/nanovllm/config.py +++ b/nanovllm/config.py @@ -3,7 +3,7 @@ from dataclasses import dataclass from transformers import AutoConfig -@dataclass +@dataclass(slots=True) class Config: model: str max_num_batched_tokens: int = 16384 diff --git a/nanovllm/sampling_params.py b/nanovllm/sampling_params.py index c9f872f..3e46598 100644 --- a/nanovllm/sampling_params.py +++ b/nanovllm/sampling_params.py @@ -1,7 +1,7 @@ from dataclasses import dataclass -@dataclass +@dataclass(slots=True) class SamplingParams: temperature: float = 1.0 max_tokens: int = 64 diff --git a/nanovllm/utils/context.py b/nanovllm/utils/context.py index 2281888..3b02a1d 100644 --- a/nanovllm/utils/context.py +++ b/nanovllm/utils/context.py @@ -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