Sana icon indicating copy to clipboard operation
Sana copied to clipboard

Published 20 hours ago • verified publisher icon NVlabs

Does SANA-Sprint not support Xformer?

Open hungdche opened this issue 7 months ago • 2 comments

Thank you for releasing SANA-Sprint code!

It seems like SANA-Sprint doesn't support xformer memory efficient attention? https://github.com/NVlabs/Sana/blob/d88cb447bf83c73d72ce084a215db5968c4cf2cf/app/sana_sprint_pipeline.py#L28

I am wondering if this is correct from my understanding, and if so, is it trivial to enable it? Thanks!

hungdche avatar Mar 24 '25 15:03 hungdche