RUCAIBox/TextBox
A unified Python/PyTorch library for text generation using 47 pre-trained language models across 13 task types.

Velocity · 7d
+0.5
★ / day
Trend
→steady
star history
TextBox 2.0 provides a standardized framework for applying pre-trained language models to text generation tasks including translation, story generation, and style transfer. It incorporates 47 pre-trained models covering general, translation, Chinese, dialogue, controllable, distilled, prompting, and lightweight categories. The library supports 4 pre-training objectives and efficient training strategies such as distributed data parallel.