TensorRT-LLMs/tensorrt_llm/scaffolding/contrib/__init__.py
WeiHaocheng 3fc2a16920
feat(part 2): Enhance the integrated robustness of scaffolding with __init__.py #3305 (#3731)
Signed-off-by: fredw (generated by with_the_same_user script) <20514172+WeiHaocheng@users.noreply.github.com>
2025-04-24 18:47:03 +08:00

13 lines
321 B
Python

from tensorrt_llm.scaffolding import * # noqa
from .AsyncGeneration import StreamGenerationTask, stream_generation_handler
from .Dynasor import DynasorGenerationController
__all__ = [
# AsyncGeneration
"stream_generation_handler",
"StreamGenerationTask",
# Dynasor
"DynasorGenerationController",
]