You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
langchain/libs/partners/nvidia-trt/tests/unit_tests/test_llms.py

8 lines
256 B
Python

"""Test TritonTensorRT Chat API wrapper."""
from langchain_nvidia_trt import TritonTensorRTLLM
def test_initialization() -> None:
"""Test integration initialization."""
TritonTensorRTLLM(model_name="ensemble", server_url="http://localhost:8001")