Memory Leak

[deepbeepmeep]

Thanks for this very interesting work that at last allows us to have some level of direction in video generation.
First run works quite well (less than 40s on a RTX 4090 24GB) but successive runs are very slow.
There is obviously a memory leak towards the end of the generation process as GPU memory is never released.

[deepbeepmeep]

Ok I found a fix, it is the VAE that clutters the GPU memory.

In file "demo/motiondirectory.py"
Add the following just after line 155 ( device ="cuda")

self.pipe.enable_vae_slicing()

[ruizhaocv]

Thanks a lot. The demo script has been updated to fix this issue.