From 9989f78e144a925d083af181844f74ff63d9d3bc Mon Sep 17 00:00:00 2001 From: Mikhail Khludnev Date: Tue, 20 Feb 2024 00:27:36 +0300 Subject: [PATCH] fix --- libs/partners/nvidia-trt/langchain_nvidia_trt/llms.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/libs/partners/nvidia-trt/langchain_nvidia_trt/llms.py b/libs/partners/nvidia-trt/langchain_nvidia_trt/llms.py index d156f9e6fbd67..111ef82bff469 100644 --- a/libs/partners/nvidia-trt/langchain_nvidia_trt/llms.py +++ b/libs/partners/nvidia-trt/langchain_nvidia_trt/llms.py @@ -383,7 +383,7 @@ def __init__( self, llm: TritonTensorRTLLM, request_id: str, - force_batch: bool, + signal_stop: bool, stop_words: Sequence[str], ) -> None: """Instantiate the generator class."""