Commit Graph

5 Commits

Author SHA1 Message Date
Yexiong Lin
36d6d91b90 update text2video.py 2025-03-04 19:54:56 +11:00
Yexiong Lin
1c7b73d13e Add the support for fp8 t5 2025-03-04 19:54:56 +11:00
Yexiong Lin
db54b7c613 Update README.md and text2video.py to offload model and enable using fp8 2025-03-04 19:54:56 +11:00
Adrian Corduneanu
0e3c42a830
Update text2video.py to reduce GPU memory by emptying cache (#44)
* Update text2video.py to reduce GPU memory by emptying cache

If offload_model is set, empty_cache() must be called after the model is moved to CPU to actually free the GPU. I verified on a RTX 4090 that without calling empty_cache the model remains in memory and the subsequent vae decoding never finishes.

* Update text2video.py only one empty_cache needed before vae decode
2025-02-26 18:56:57 +08:00
WanX-Video-1
65386b2e03 init upload 2025-02-25 22:07:47 +08:00