Update flash-attn version for compatibility

openai / consistency_models

Official repo for consistency models.

MIT License

6.08k stars 411 forks source link

Update flash-attn version for compatibility #13

Closed ashutosh1919 closed 1 year ago

ashutosh1919 commented 1 year ago

FlashAttention() class doesn't contain device parameter in constructor in the latest version. The latest version which has this parameter is 0.2.8.