torch import missed in example
Browse files`import torch` is required for `torch.bfloat16` in "How to get Started" section example
README.md
CHANGED
|
@@ -63,6 +63,7 @@ You can load the model as follows.
|
|
| 63 |
|
| 64 |
```python
|
| 65 |
from transformers import AutoProcessor, AutoModelForImageTextToText
|
|
|
|
| 66 |
|
| 67 |
model_path = "HuggingFaceTB/SmolVLM2-500M-Video-Instruct"
|
| 68 |
processor = AutoProcessor.from_pretrained(model_path)
|
|
|
|
| 63 |
|
| 64 |
```python
|
| 65 |
from transformers import AutoProcessor, AutoModelForImageTextToText
|
| 66 |
+
import torch
|
| 67 |
|
| 68 |
model_path = "HuggingFaceTB/SmolVLM2-500M-Video-Instruct"
|
| 69 |
processor = AutoProcessor.from_pretrained(model_path)
|