From 5ed704ec8c4e68f1bc846ab4e3c9e355585d62da Mon Sep 17 00:00:00 2001 From: Massimiliano Pronesti Date: Sun, 4 Feb 2024 03:17:55 +0100 Subject: [PATCH] docs: fix langchain (#2736) --- docs/source/serving/serving_with_langchain.rst | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/docs/source/serving/serving_with_langchain.rst b/docs/source/serving/serving_with_langchain.rst index 2e1ce688290a..6440c8aad598 100644 --- a/docs/source/serving/serving_with_langchain.rst +++ b/docs/source/serving/serving_with_langchain.rst @@ -9,13 +9,13 @@ To install langchain, run .. code-block:: console - $ pip install langchain -q + $ pip install langchain langchain_community -q To run inference on a single or multiple GPUs, use ``VLLM`` class from ``langchain``. .. code-block:: python - from langchain.llms import VLLM + from langchain_community.llms import VLLM llm = VLLM(model="mosaicml/mpt-7b", trust_remote_code=True, # mandatory for hf models @@ -28,4 +28,4 @@ To run inference on a single or multiple GPUs, use ``VLLM`` class from ``langcha print(llm("What is the capital of France ?")) -Please refer to this `Tutorial `_ for more details. \ No newline at end of file +Please refer to this `Tutorial `_ for more details.