Tiny LM - a trl-internal-testing Collection

trl-internal-testing 's Collections

Tiny reward models

Testing datasets

Tiny LM

updated Mar 2

trl-internal-testing/tiny-BloomForCausalLM

Text Generation • 2.01M • Updated Sep 24, 2025 • 91k
trl-internal-testing/tiny-CohereForCausalLM

Text Generation • 2.05M • Updated May 15 • 37.4k
trl-internal-testing/tiny-FalconMambaForCausalLM

Text Generation • 1.05M • Updated Sep 22 • 19.1k
trl-internal-testing/tiny-GemmaForCausalLM

Text Generation • 2.05M • Updated Sep 23 • 27.1k
trl-internal-testing/tiny-Gemma2ForCausalLM

Text Generation • 2.05M • Updated Sep 23 • 49.8k • 1
trl-internal-testing/tiny-GPT2LMHeadModel

Text Generation • 412k • Updated Dec 19, 2025 • 44.8k
trl-internal-testing/tiny-GPTNeoXForCausalLM

Text Generation • 807k • Updated Sep 8 • 41.7k
trl-internal-testing/tiny-LlamaForCausalLM-3

Text Generation • 2.05M • Updated Sep 8 • 25.4k
trl-internal-testing/tiny-LlamaForCausalLM-3.1

Text Generation • 2.05M • Updated Sep 8 • 25.8k
trl-internal-testing/tiny-LlamaForCausalLM-3.2

Text Generation • 1.03M • Updated Sep 10 • 91.7k • 1
trl-internal-testing/tiny-MistralForCausalLM-0.1

Text Generation • 514k • Updated Sep 15 • 21.6k
trl-internal-testing/tiny-MistralForCausalLM-0.2

Text Generation • 514k • Updated Sep 15 • 59.8k
trl-internal-testing/tiny-OPTForCausalLM

Text Generation • 420k • Updated Sep 23 • 7.26k
trl-internal-testing/tiny-Phi3ForCausalLM

Text Generation • 514k • Updated Apr 14 • 58.1k
trl-internal-testing/tiny-Qwen2ForCausalLM-2.5

Text Generation • 2.44M • Updated Sep 8 • 2.76M • 59
trl-internal-testing/tiny-Qwen2ForCausalLM-2.5-Coder

Text Generation • 1.22M • Updated Sep 10 • 2.92k
trl-internal-testing/tiny-Qwen3ForCausalLM

Text Generation • 2.46M • Updated Sep 8 • 814k • 12

Note ^ Decoder-only
trl-internal-testing/tiny-BartModel

Feature Extraction • 3.11M • Updated Dec 19, 2025 • 8.15k
trl-internal-testing/tiny-T5ForConditionalGeneration

3.61M • Updated Dec 19, 2025 • 37.9k

Note ^ Encoder-decoder