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)
|