fix: flashfftconv imports

togethercomputer / stripedhyena

Repository for StripedHyena, a state-of-the-art beyond Transformer architecture

Apache License 2.0

299 stars 21 forks source link

Closed stereoplegic closed 9 months ago

stereoplegic commented 9 months ago

src/model.py:
- added conditional import of FlashDepthwiseConv1D (re: #3) with try/except
- fixed FlashFFTConv conditional import, add try/except
requirements.txt:
- added flashfftconv (from git, commented out for now)

Zymrael commented 9 months ago

Thanks!