mirror of
https://github.com/NVIDIA/TensorRT-LLM.git
synced 2026-01-14 06:27:45 +08:00
45 lines
1.5 KiB
Python
45 lines
1.5 KiB
Python
# Copyright (c) 2025, NVIDIA CORPORATION. All rights reserved.
|
|
|
|
from .benchmark import ScaffoldingBenchRequest, async_scaffolding_benchmark
|
|
from .contrib.TreeInference.tree_controllers import (MCTSController,
|
|
TOTController)
|
|
from .controller import (BestOfNController, Controller, MajorityVoteController,
|
|
NativeGenerationController, NativeRewardController,
|
|
ParallelProcess, PRMController)
|
|
from .math_utils import (extract_answer_from_boxed, extract_answer_with_regex,
|
|
get_digit_majority_vote_result)
|
|
from .scaffolding_llm import ScaffoldingLlm
|
|
from .task import GenerationTask, RewardTask, Task, TaskStatus
|
|
from .task_collection import (GenerationTokenCounter, TaskCollection,
|
|
with_task_collection)
|
|
from .worker import OpenaiWorker, TRTLLMWorker, TRTOpenaiWorker, Worker
|
|
|
|
__all__ = [
|
|
"ScaffoldingLlm",
|
|
"ParallelProcess",
|
|
"Controller",
|
|
"NativeGenerationController",
|
|
"NativeRewardController",
|
|
"PRMController",
|
|
"MajorityVoteController",
|
|
"BestOfNController",
|
|
"MCTSController",
|
|
"TOTController",
|
|
"Task",
|
|
"GenerationTask",
|
|
"RewardTask",
|
|
"Worker",
|
|
"OpenaiWorker",
|
|
"TRTOpenaiWorker",
|
|
"TRTLLMWorker",
|
|
"TaskStatus",
|
|
"extract_answer_from_boxed",
|
|
"extract_answer_with_regex",
|
|
"get_digit_majority_vote_result",
|
|
"TaskCollection",
|
|
"with_task_collection",
|
|
"GenerationTokenCounter",
|
|
"async_scaffolding_benchmark",
|
|
"ScaffoldingBenchRequest",
|
|
]
|