Update README.md
Browse files
README.md
CHANGED
|
@@ -192,6 +192,7 @@ An OpenAI-compatible API will be available at `http://localhost:8000/v1`.
|
|
| 192 |
|
| 193 |
Or you can refer to the following Python script for offline inference
|
| 194 |
```python
|
|
|
|
| 195 |
from vllm import LLM, SamplingParams
|
| 196 |
from transformers import AutoTokenizer
|
| 197 |
|
|
|
|
| 192 |
|
| 193 |
Or you can refer to the following Python script for offline inference
|
| 194 |
```python
|
| 195 |
+
import torch
|
| 196 |
from vllm import LLM, SamplingParams
|
| 197 |
from transformers import AutoTokenizer
|
| 198 |
|