torch import missed in example (#9)
Browse files- torch import missed in example (e85be16a7bb2c3c189881a5038df9dbf691246ea)
Co-authored-by: Ahamed Musthafa R S <[email protected]>
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)
|