31 lines
895 B
ReStructuredText
31 lines
895 B
ReStructuredText
.. _run_on_langchain:
|
|
|
|
Serving with Langchain
|
|
============================
|
|
|
|
vLLM is also available via `Langchain <https://github.com/langchain-ai/langchain>`_ .
|
|
|
|
To install langchain, run
|
|
|
|
.. code-block:: console
|
|
|
|
$ pip install langchain -q
|
|
|
|
To run inference on a single or multiple GPUs, use ``VLLM`` class from ``langchain``.
|
|
|
|
.. code-block:: python
|
|
|
|
from langchain.llms import VLLM
|
|
|
|
llm = VLLM(model="mosaicml/mpt-7b",
|
|
trust_remote_code=True, # mandatory for hf models
|
|
max_new_tokens=128,
|
|
top_k=10,
|
|
top_p=0.95,
|
|
temperature=0.8,
|
|
# tensor_parallel_size=... # for distributed inference
|
|
)
|
|
|
|
print(llm("What is the capital of France ?"))
|
|
|
|
Please refer to this `Tutorial <https://github.com/langchain-ai/langchain/blob/master/docs/extras/integrations/llms/vllm.ipynb>`_ for more details. |