Update README.md
Browse files
README.md
CHANGED
|
@@ -28,6 +28,7 @@ You will also need to install [flash-attn](https://github.com/Dao-AILab/flash-at
|
|
| 28 |
Please note that you need at least 80GB VRAM to run this pipeline. CPU offloading is having issues at the moment (PRs welcome!).
|
| 29 |
|
| 30 |
```python
|
|
|
|
| 31 |
from hyvideo.diffusion.pipelines.pipeline_hunyuan_video import HunyuanVideoPipeline
|
| 32 |
from hyvideo.modules.models import HYVideoDiffusionTransformer
|
| 33 |
from hyvideo.vae.autoencoder_kl_causal_3d import AutoencoderKLCausal3D
|
|
|
|
| 28 |
Please note that you need at least 80GB VRAM to run this pipeline. CPU offloading is having issues at the moment (PRs welcome!).
|
| 29 |
|
| 30 |
```python
|
| 31 |
+
import os
|
| 32 |
from hyvideo.diffusion.pipelines.pipeline_hunyuan_video import HunyuanVideoPipeline
|
| 33 |
from hyvideo.modules.models import HYVideoDiffusionTransformer
|
| 34 |
from hyvideo.vae.autoencoder_kl_causal_3d import AutoencoderKLCausal3D
|