haoliuhl / ringattention

Transformers with Arbitrarily Large Context
Apache License 2.0
630 stars 50 forks source link

Incorrect project requirements #16

Closed hadipash closed 4 months ago

hadipash commented 7 months ago

In the project requirements, it is specified that the version of jax is 0.4.13. However, Pallas was added in the version 0.4.16 (https://github.com/google/jax/commit/d872812a359a3bafcfdeba1fcdb874ec77c209db).

haoliuhl commented 4 months ago

Updated! Thanks!