28 lines
719 B
ReStructuredText
28 lines
719 B
ReStructuredText
|
.. _run_on_llamaindex:
|
||
|
|
|
||
|
|
Serving with llama_index
|
||
|
|
============================
|
||
|
|
|
||
|
|
vLLM is also available via `llama_index <https://github.com/run-llama/llama_index>`_ .
|
||
|
|
|
||
|
|
To install llamaindex, run
|
||
|
|
|
||
|
|
.. code-block:: console
|
||
|
|
|
||
|
|
$ pip install llama-index-llms-vllm -q
|
||
|
|
|
||
|
|
To run inference on a single or multiple GPUs, use ``Vllm`` class from ``llamaindex``.
|
||
|
|
|
||
|
|
.. code-block:: python
|
||
|
|
|
||
|
|
from llama_index.llms.vllm import Vllm
|
||
|
|
|
||
|
|
llm = Vllm(
|
||
|
|
model="microsoft/Orca-2-7b",
|
||
|
|
tensor_parallel_size=4,
|
||
|
|
max_new_tokens=100,
|
||
|
|
vllm_kwargs={"swap_space": 1, "gpu_memory_utilization": 0.5},
|
||
|
|
)
|
||
|
|
|
||
|
|
Please refer to this `Tutorial <https://docs.llamaindex.ai/en/latest/examples/llm/vllm/>`_ for more details.
|