TinyMoE-100M-2x16

TinyMoE-100M-2x16 is a compact, highly efficient Sparse Mixture of Experts (MoE) language model built upon the Mixtral/Mistral architecture. Designed for research, edge applications, and resource-constrained environments, this model leverages an expert-routing mechanism to balance a larger total parameter capacity with ultra-low computational overhead during inference.

Model Details

  • Architecture: Sparse Mixture of Experts (MoE)
  • Total Parameters: ~200m total parameters
  • Active Parameters per Token: ~50M active parameters
  • Expert Configuration: 16 total local experts, 2 active experts routed per token (num_experts_per_tok": 2)
  • Context Length: 1024 tokens
  • Base Architecture: Mixtral / Mistral For Causal LM
  • License: MIT

Training Data

This model was trained onFineWeb-Edu-Dedup 60% and Cosmopedia-v2 40%.

image image image image image

Quick Start

You can load and experiment with this model using the Hugging Face transformers library:

from transformers import AutoModelForCausalLM, AutoTokenizer

model_id = "FlameF0X/TinyMoE-100m-2x16"
tokenizer = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForCausalLM.from_pretrained(model_id)

input_text = "Wikipedia is a free"
inputs = tokenizer(input_text, return_tensors="pt")

outputs = model.generate(**inputs, max_new_tokens=50)
print(tokenizer.decode(outputs[0], skip_special_tokens=True))
Downloads last month
246
Safetensors
Model size
0.2B params
Tensor type
F32
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Dataset used to train FlameF0X/TinyMoE-200m-2x16

Space using FlameF0X/TinyMoE-200m-2x16 1

Collection including FlameF0X/TinyMoE-200m-2x16