Update Readme to have all imports in python example

#5
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"