mirror of
https://github.com/NVIDIA/TensorRT-LLM.git
synced 2026-01-13 22:18:36 +08:00
14 lines
329 B
Python
14 lines
329 B
Python
from tensorrt_llm.llmapi.llm import _TorchLLM
|
|
|
|
|
|
class LLM(_TorchLLM):
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
raise ImportError(
|
|
"_torch.llm is deprecated, please use `from tensorrt_llm import LLM` directly"
|
|
)
|
|
|
|
|
|
# Keep the LLM class to guide the users to use the default LLM class
|
|
__all__ = ['LLM']
|