Text Generation
Transformers
Safetensors
English
qwen3
littlelearner
bounded
base
text-generation-inference
Instructions to use littlelearner/littlelearner-5b-base with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use littlelearner/littlelearner-5b-base with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-generation", model="littlelearner/littlelearner-5b-base")# Load model directly from transformers import AutoTokenizer, AutoModelForCausalLM tokenizer = AutoTokenizer.from_pretrained("littlelearner/littlelearner-5b-base") model = AutoModelForCausalLM.from_pretrained("littlelearner/littlelearner-5b-base", device_map="auto") - Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- vLLM
How to use littlelearner/littlelearner-5b-base with vLLM:
Install from pip and serve model
# Install vLLM from pip: pip install vllm # Start the vLLM server: vllm serve "littlelearner/littlelearner-5b-base" # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:8000/v1/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "littlelearner/littlelearner-5b-base", "prompt": "Once upon a time,", "max_tokens": 512, "temperature": 0.5 }'Use Docker
docker model run hf.co/littlelearner/littlelearner-5b-base
- SGLang
How to use littlelearner/littlelearner-5b-base with SGLang:
Install from pip and serve model
# Install SGLang from pip: pip install sglang # Start the SGLang server: python3 -m sglang.launch_server \ --model-path "littlelearner/littlelearner-5b-base" \ --host 0.0.0.0 \ --port 30000 # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:30000/v1/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "littlelearner/littlelearner-5b-base", "prompt": "Once upon a time,", "max_tokens": 512, "temperature": 0.5 }'Use Docker images
docker run --gpus all \ --shm-size 32g \ -p 30000:30000 \ -v ~/.cache/huggingface:/root/.cache/huggingface \ --env "HF_TOKEN=<secret>" \ --ipc=host \ lmsysorg/sglang:latest \ python3 -m sglang.launch_server \ --model-path "littlelearner/littlelearner-5b-base" \ --host 0.0.0.0 \ --port 30000 # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:30000/v1/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "littlelearner/littlelearner-5b-base", "prompt": "Once upon a time,", "max_tokens": 512, "temperature": 0.5 }' - Docker Model Runner
How to use littlelearner/littlelearner-5b-base with Docker Model Runner:
docker model run hf.co/littlelearner/littlelearner-5b-base
Model card: add minimal transformers + vLLM usage snippets
Browse files
README.md
CHANGED
|
@@ -21,21 +21,30 @@ Part of the **LittleLearner** scale-up study (*pedagogically-controlled knowledg
|
|
| 21 |
## Model
|
| 22 |
- **Architecture:** Qwen3 dense (`Qwen3ForCausalLM`) — standard `transformers`, no custom code / `trust_remote_code`.
|
| 23 |
- **Size:** 5.04B params — hidden 3072, 44 layers, 24 query / 8 KV heads, FFN 9216. **Context:** 4096.
|
| 24 |
-
- **Tokenizer:** custom 64k byte-level BPE with per-digit splitting (
|
| 25 |
-
- **Pretraining:** 88B tokens on K-5 **LittleCurriculum** (FineWeb-Edu filtered to U.S. grades K–5). WSD schedule, sharded Muon
|
| 26 |
|
| 27 |
## Evaluation
|
| 28 |
- In-domain bits-per-byte (BPB): **0.536** (vs the 2B nanochat reference 0.805).
|
| 29 |
|
| 30 |
-
|
| 31 |
## Usage
|
| 32 |
```python
|
|
|
|
| 33 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 34 |
-
|
| 35 |
-
tok = AutoTokenizer.from_pretrained(
|
| 36 |
-
model = AutoModelForCausalLM.from_pretrained(
|
| 37 |
ids = tok("The sum of 2 and 3 is", return_tensors="pt").to(model.device)
|
| 38 |
-
print(tok.decode(model.generate(**ids, max_new_tokens=64)[0]))
|
| 39 |
```
|
| 40 |
|
| 41 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 21 |
## Model
|
| 22 |
- **Architecture:** Qwen3 dense (`Qwen3ForCausalLM`) — standard `transformers`, no custom code / `trust_remote_code`.
|
| 23 |
- **Size:** 5.04B params — hidden 3072, 44 layers, 24 query / 8 KV heads, FFN 9216. **Context:** 4096.
|
| 24 |
+
- **Tokenizer:** custom 64k byte-level BPE with per-digit splitting (ChatML special tokens).
|
| 25 |
+
- **Pretraining:** 88B tokens on K-5 **LittleCurriculum** (FineWeb-Edu filtered to U.S. grades K–5). WSD schedule, sharded Muon, MXFP8, Megatron-Core on 8×B200.
|
| 26 |
|
| 27 |
## Evaluation
|
| 28 |
- In-domain bits-per-byte (BPB): **0.536** (vs the 2B nanochat reference 0.805).
|
| 29 |
|
|
|
|
| 30 |
## Usage
|
| 31 |
```python
|
| 32 |
+
# transformers (completion)
|
| 33 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 34 |
+
repo = "manueldeprada/littlelearner-5b-bounded-base"
|
| 35 |
+
tok = AutoTokenizer.from_pretrained(repo)
|
| 36 |
+
model = AutoModelForCausalLM.from_pretrained(repo, dtype="bfloat16", device_map="cuda")
|
| 37 |
ids = tok("The sum of 2 and 3 is", return_tensors="pt").to(model.device)
|
| 38 |
+
print(tok.decode(model.generate(**ids, max_new_tokens=64)[0], skip_special_tokens=True))
|
| 39 |
```
|
| 40 |
|
| 41 |
+
```python
|
| 42 |
+
# vLLM
|
| 43 |
+
from vllm import LLM, SamplingParams
|
| 44 |
+
llm = LLM("manueldeprada/littlelearner-5b-bounded-base")
|
| 45 |
+
print(llm.generate(["The sum of 2 and 3 is"], SamplingParams(max_tokens=64))[0].outputs[0].text)
|
| 46 |
+
```
|
| 47 |
+
|
| 48 |
+
> **vLLM note:** vLLM JIT-compiles CUDA kernels, so it needs a CUDA toolkit (`nvcc`) on PATH. On an HPC cluster, `module load cuda/13.x cudnn/9.x` (match your torch CUDA build). On a box with no toolkit, pass `LLM(repo, enforce_eager=True)`. Verified on vLLM 0.15 and 0.23 (and transformers ≥ 4.51).
|
| 49 |
+
|
| 50 |
+
> Research artifact. The **bounded** models carry an intentional K–5 knowledge boundary (they cannot model above-grade-5 material). Base models are **not** instruction-tuned.
|