TensorRT-LLMs/tensorrt_llm/auto_parallel/__init__.py
2024-04-16 19:40:08 +08:00

10 lines
217 B
Python

from .auto_parallel import auto_parallel
from .cluster_info import infer_cluster_config
from .config import AutoParallelConfig
__all__ = [
'auto_parallel',
'AutoParallelConfig',
'infer_cluster_config',
]