mirror of
https://github.com/NVIDIA/TensorRT-LLM.git
synced 2026-01-14 06:27:45 +08:00
8 lines
109 B
Python
8 lines
109 B
Python
from dataclasses import dataclass
|
|
|
|
|
|
@dataclass
|
|
class Pool(object):
|
|
num_kv_heads: int
|
|
num_layers: int
|