First Hugging Face integration of minGRU models from the paper «Were RNNs All We Needed?». This model uses GPT-2 tokenizer and trained on roneneldan/TinyStories dataset. Make sure you have installed «minGRU-pytorch» library by running «pip install minGRU-pytorch». I am thankful to Leo Feng, Frederick Tung, Mohamed Osama Ahmed, Yoshua Bengio and Hossein Hajimirsadeghi for their papers.
Модальности:
Генерация текста
Задача: Генерация текста
Автор: suayptalha
Теги: mingru, custom_code, minGRU, hf_integration, en
Лайков: 4 | Загрузок: 15
Описание основано на материалах HuggingFace. Перевод выполнен автоматически.