Added flash attention.

[mmcdermott]

No description provided.

[codecov]

Codecov Report

Attention: 4 lines in your changes are missing coverage. Please review.

Comparison is base (834c850) 86.04% compared to head (f479082) 86.12%.
Report is 13 commits behind head on dev.

Additional details and impacted files

@@            Coverage Diff             @@
##              dev      #70      +/-   ##
==========================================
+ Coverage   86.04%   86.12%   +0.07%     
==========================================
  Files          34       34              
  Lines        6401     6408       +7     
==========================================
+ Hits         5508     5519      +11     
+ Misses        893      889       -4     

Files	Coverage Δ
EventStream/data/pytorch_dataset.py	89.13% <100.00%> (+2.24%)	⬆️
...Stream/evaluation/general_generative_evaluation.py	25.67% <100.00%> (ø)
...tream/transformer/lightning_modules/fine_tuning.py	83.91% <100.00%> (ø)
...ansformer/lightning_modules/generative_modeling.py	89.69% <100.00%> (ø)
EventStream/transformer/config.py	88.94% <0.00%> (ø)
EventStream/transformer/transformer.py	87.30% <88.00%> (-0.26%)	⬇️

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

[mmcdermott]

Moving to the NeurIPS v0.5 release.