A better fix for #3 to support new `flash-attn` versions

openai / consistency_models

Official repo for consistency models.

MIT License

6.17k stars 420 forks source link

Open ain-soph opened 1 year ago

ain-soph commented 1 year ago

Fix #3 in a better way to support newer version of flash-attn.

ShashwatNigam99 commented 1 year ago

Thanks for this concise fix for the flash-attn issue. Works!