OUT OF MEMORY - Githubissues

wooyeolBaek / attention-map

🚀 Cross attention map tools for huggingface/diffusers

MIT License

150 stars 9 forks source link

Open chenbinghui1 opened 5 months ago

chenbinghui1 commented 5 months ago

When I replace the attn processor with this project, I find the used GPU memory will be increased. Who knows the reason?

wooyeolBaek commented 5 months ago

Have you tried torch.cuda.empty_cache() ?