CUDA out of memory

[seulqxq]

Hello, I ran python app.py and encountered an out-of-memory error after Flow Sampling. What could be causing this issue? I noticed the README states that 16GB should be sufficient on Windows.

[ashawkey]

@seulqxq Hi, this seems strange. I have updated the code to reduce VAE memory usage, maybe you could try again.
Also, you may check the torch version and installation (if it's using memory-efficient flash-attention?), or explicitly install and use flash-attn.