Gated Linear Attention Transformers with Hardware-Efficient Training
Paper • 2312.06635 • Published • 9
A 340M parameter GLA (Gated Linear Attention) model.
from transformers import AutoModelForCausalLM, AutoTokenizer
model = AutoModelForCausalLM.from_pretrained("Zoha4/cortx-340M-1.3B-2k")
tokenizer = AutoTokenizer.from_pretrained("Zoha4/cortx-340M-1.3B-2k")
Yang et al. Gated Linear Attention Transformers with Hardware-Efficient Training. ICML, 2024. PDF
Zellers et al. HellaSwag: Can a Machine Really Finish Your Sentence? In ACL, 2019. PDF
Penedo et al. The FineWeb Datasets: Decanting the Web for the Finest Text Data at Scale, 2024. Blog · Fineweb-Edu