shibing624/textgen
TextGen is a Python library implementing various text generation models including LLaMA, ChatGLM, GPT2, BART, and T5 for training and inference.

Velocity · 7d
+0.5
★ / day
Trend
→steady
star history
This repository provides implementations of multiple text generation models including LLaMA, ChatGLM, BLOOM, GPT2, Seq2Seq, BART, T5, and SongNet. It offers both training and inference capabilities with multi-GPU support and techniques like NEFTune for fine-tuning. The library aims to be plug-and-play for developers working on text generation tasks.