mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2025-12-09 23:15:31 +08:00
[doc] add missing import (#18133)
Signed-off-by: reidliu41 <reid201711@gmail.com> Co-authored-by: reidliu41 <reid201711@gmail.com>
This commit is contained in:
parent
d62a076e84
commit
9ccc6ded42
@ -74,6 +74,8 @@ Tensor parallelism (`tensor_parallel_size` option) can be used to split the mode
|
|||||||
The following code splits the model across 2 GPUs.
|
The following code splits the model across 2 GPUs.
|
||||||
|
|
||||||
```python
|
```python
|
||||||
|
from vllm import LLM
|
||||||
|
|
||||||
llm = LLM(model="ibm-granite/granite-3.1-8b-instruct",
|
llm = LLM(model="ibm-granite/granite-3.1-8b-instruct",
|
||||||
tensor_parallel_size=2)
|
tensor_parallel_size=2)
|
||||||
```
|
```
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user