amrs-tech commited on
Commit
e85be16
·
verified ·
1 Parent(s): 3444947

torch import missed in example

Browse files

`import torch` is required for `torch.bfloat16` in "How to get Started" section example

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)