torch import missed in example

#9
Files changed (1) hide show
  1. README.md +1 -0
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)