Tiny models - a trl-internal-testing Collection

trl-internal-testing 's Collections

Tiny reward models

Tiny models

updated 28 days ago

trl-internal-testing/tiny-BloomForCausalLM

Text Generation • Updated Nov 25, 2024 • 37.7k
trl-internal-testing/tiny-CohereForCausalLM

Text Generation • Updated Nov 25, 2024 • 37.2k
trl-internal-testing/tiny-DbrxForCausalLM

Text Generation • Updated Nov 25, 2024 • 37.6k
trl-internal-testing/tiny-FalconMambaForCausalLM

Text Generation • Updated Nov 25, 2024 • 37.5k
trl-internal-testing/tiny-GemmaForCausalLM

Text Generation • Updated Nov 25, 2024 • 37.7k
trl-internal-testing/tiny-Gemma2ForCausalLM

Text Generation • Updated Nov 25, 2024 • 39k • 1
trl-internal-testing/tiny-GPT2LMHeadModel

Text Generation • Updated Nov 25, 2024 • 41.9k
trl-internal-testing/tiny-GPTNeoXForCausalLM

Text Generation • Updated Nov 25, 2024 • 37.6k
trl-internal-testing/tiny-LlamaForCausalLM-3

Text Generation • Updated Nov 25, 2024 • 37.7k
trl-internal-testing/tiny-LlamaForCausalLM-3.1

Text Generation • Updated Nov 25, 2024 • 37.7k
trl-internal-testing/tiny-LlamaForCausalLM-3.2

Text Generation • Updated Nov 25, 2024 • 429k • 1
trl-internal-testing/tiny-MistralForCausalLM-0.1

Text Generation • Updated Nov 25, 2024 • 37.7k
trl-internal-testing/tiny-MistralForCausalLM-0.2

Text Generation • Updated Nov 25, 2024 • 86.6k
trl-internal-testing/tiny-OPTForCausalLM

Text Generation • Updated Nov 25, 2024 • 37.8k
trl-internal-testing/tiny-Phi3ForCausalLM

Text Generation • Updated Nov 25, 2024 • 37.7k
trl-internal-testing/tiny-Qwen2ForCausalLM-2.5

Text Generation • Updated Nov 25, 2024 • 1.23M • 1
trl-internal-testing/tiny-Qwen2ForCausalLM-2.5-Coder

Text Generation • Updated Nov 25, 2024 • 9
trl-internal-testing/tiny-Qwen3ForCausalLM

Text Generation • Updated 30 days ago • 4

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

Feature Extraction • Updated Nov 25, 2024 • 39.5k
trl-internal-testing/tiny-T5ForConditionalGeneration

Text2Text Generation • Updated Nov 25, 2024 • 309k

Note ^ Encoder-decoder
trl-internal-testing/tiny-Idefics2ForConditionalGeneration

Image-Text-to-Text • Updated Nov 25, 2024 • 5.79k
trl-internal-testing/tiny-LlavaForConditionalGeneration

Image-Text-to-Text • Updated Nov 25, 2024 • 12.2k
trl-internal-testing/tiny-LlavaNextForConditionalGeneration

Image-Text-to-Text • Updated Nov 25, 2024 • 7.09k
trl-internal-testing/tiny-PaliGemmaForConditionalGeneration

Image-Text-to-Text • Updated Nov 25, 2024 • 5

Note ^ Vision language models