mirror of
https://github.com/Wan-Video/Wan2.1.git
synced 2025-06-05 23:04:54 +00:00
* 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 |
||
---|---|---|
.. | ||
configs | ||
distributed | ||
modules | ||
utils | ||
__init__.py | ||
image2video.py | ||
text2video.py |