nielsr HF staff matthiasruiz commited on
Commit
4356641
1 Parent(s): 6774b71

Update Readme to have all imports in python example (#5)

Browse files

- Update Readme to have all imports in python example (b772464536512d7a6a9d30420916c4b1c28894f9)


Co-authored-by: Matthias Ruiz <[email protected]>

Files changed (1) hide show
  1. README.md +2 -0
README.md CHANGED
@@ -56,6 +56,8 @@ Below is an example script to run generation in `float16` precision on a GPU dev
56
  ```python
57
  import av
58
  import torch
 
 
59
  from transformers import LlavaNextVideoProcessor, LlavaNextVideoForConditionalGeneration
60
 
61
  model_id = "llava-hf/LLaVA-NeXT-Video-7B-hf"
 
56
  ```python
57
  import av
58
  import torch
59
+ import numpy as np
60
+ from huggingface_hub import hf_hub_download
61
  from transformers import LlavaNextVideoProcessor, LlavaNextVideoForConditionalGeneration
62
 
63
  model_id = "llava-hf/LLaVA-NeXT-Video-7B-hf"