Tiny models - a trl-internal-testing Collection

trl-internal-testing 's Collections

Tiny reward models

Tiny models

updated 6 days ago

trl-internal-testing/tiny-BloomForCausalLM

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

Text Generation • Updated 8 days ago • 1

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

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

Text2Text Generation • Updated Nov 25, 2024 • 279k

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

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

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

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

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

Note ^ Vision language models