Update Readme to have all imports in python example
#5
by matthiasruiz - opened
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"
|