← all repositories

RUCAIBox/TextBox

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

1.1k stars Python Language ModelsML Frameworks
TextBox
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.

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