[feat] Added num_gpu_blocks limit gpu blocks.
This commit is contained in:
@@ -22,6 +22,7 @@ class Config:
|
||||
cpu_memory_gb: float = 16.0 # CPU memory limit for KV cache
|
||||
offload_policy: str = "lru" # "lru", "fifo", or full class path
|
||||
num_transfer_streams: int = 4 # Number of CUDA streams for async transfers
|
||||
num_gpu_blocks: int = -1 # User-specified GPU blocks count, -1 = auto (use max available)
|
||||
|
||||
# Computed fields for offload (set in __post_init__ or by ModelRunner)
|
||||
num_gpu_kvcache_blocks: int = -1
|
||||
|
||||
Reference in New Issue
Block a user