search

young-geng / EasyLM

Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-training, finetuning, evaluating and serving LLMs in JAX/Flax.
Apache License 2.0
2.33k stars 247 forks source link

Fix omitted dtype params in efficient memory attention function. #82

Closed Taekyoon closed 1 year ago

Taekyoon commented 1 year ago

Ref: #81

Taekyoon commented 1 year ago

Already fixed. closed :)