Zengtudor/Wan2.1
1
0
Fork 0
mirror of https://github.com/Wan-Video/Wan2.1.git synced 2025-11-04 14:16:57 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity
370f18a36d
Wan2.1/wan
History
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
..
configs init upload 2025-02-25 22:07:47 +08:00
distributed init upload 2025-02-25 22:07:47 +08:00
modules init upload 2025-02-25 22:07:47 +08:00
utils init upload 2025-02-25 22:07:47 +08:00
__init__.py init upload 2025-02-25 22:07:47 +08:00
image2video.py init upload 2025-02-25 22:07:47 +08:00
text2video.py Update text2video.py to reduce GPU memory by emptying cache (#44) 2025-02-26 18:56:57 +08:00