← all repositories

shibing624/textgen

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

979 stars Python Language ModelsML Frameworks
textgen
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.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.