Instructions to use MLVXN/MicroLLM2 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.

Libraries

How to use MLVXN/MicroLLM2 with Transformers:

# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("text-generation", model="MLVXN/MicroLLM2")
messages = [
    {"role": "user", "content": "Who are you?"},
]
pipe(messages)

# pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM

tokenizer = AutoTokenizer.from_pretrained("MLVXN/MicroLLM2")
model = AutoModelForCausalLM.from_pretrained("MLVXN/MicroLLM2", device_map="auto")
messages = [
    {"role": "user", "content": "Who are you?"},
]
inputs = tokenizer.apply_chat_template(
	messages,
	add_generation_prompt=True,
	tokenize=True,
	return_dict=True,
	return_tensors="pt",
).to(model.device)

outputs = model.generate(**inputs, max_new_tokens=256)
print(tokenizer.decode(outputs[0][inputs["input_ids"].shape[-1]:]))

Notebooks
Google Colab
Kaggle
Local Apps Settings

llama.cpp

How to use MLVXN/MicroLLM2 with llama.cpp:

Install (macOS, Linux)

curl -LsSf https://llama.app/install.sh | sh
# Start a local OpenAI-compatible server with a web UI:
llama serve -hf MLVXN/MicroLLM2:Q4_K_M
# Run inference directly in the terminal:
llama cli -hf MLVXN/MicroLLM2:Q4_K_M

Install from WinGet (Windows)

winget install llama.cpp
# Start a local OpenAI-compatible server with a web UI:
llama serve -hf MLVXN/MicroLLM2:Q4_K_M
# Run inference directly in the terminal:
llama cli -hf MLVXN/MicroLLM2:Q4_K_M

Build from source code

git clone https://github.com/ggml-org/llama.cpp.git
cd llama.cpp
cmake -B build
cmake --build build -j --target llama-app
# Start a local OpenAI-compatible server with a web UI:
./build/bin/llama serve -hf MLVXN/MicroLLM2:Q4_K_M
# Run inference directly in the terminal:
./build/bin/llama cli -hf MLVXN/MicroLLM2:Q4_K_M

Use Docker

docker model run hf.co/MLVXN/MicroLLM2:Q4_K_M

Llama new
LM Studio
Jan

vLLM

How to use MLVXN/MicroLLM2 with vLLM:

Install from pip and serve model

# Install vLLM from pip:
pip install vllm
# Start the vLLM server:
vllm serve "MLVXN/MicroLLM2"
# Call the server using curl (OpenAI-compatible API):
curl -X POST "http://localhost:8000/v1/chat/completions" \
	-H "Content-Type: application/json" \
	--data '{
		"model": "MLVXN/MicroLLM2",
		"messages": [
			{
				"role": "user",
				"content": "What is the capital of France?"
			}
		]
	}'

Use Docker

docker model run hf.co/MLVXN/MicroLLM2:Q4_K_M

SGLang

How to use MLVXN/MicroLLM2 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 "MLVXN/MicroLLM2" \
    --host 0.0.0.0 \
    --port 30000
# Call the server using curl (OpenAI-compatible API):
curl -X POST "http://localhost:30000/v1/chat/completions" \
	-H "Content-Type: application/json" \
	--data '{
		"model": "MLVXN/MicroLLM2",
		"messages": [
			{
				"role": "user",
				"content": "What is the capital of France?"
			}
		]
	}'

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 "MLVXN/MicroLLM2" \
        --host 0.0.0.0 \
        --port 30000
# Call the server using curl (OpenAI-compatible API):
curl -X POST "http://localhost:30000/v1/chat/completions" \
	-H "Content-Type: application/json" \
	--data '{
		"model": "MLVXN/MicroLLM2",
		"messages": [
			{
				"role": "user",
				"content": "What is the capital of France?"
			}
		]
	}'

Ollama
How to use MLVXN/MicroLLM2 with Ollama:
```
ollama run hf.co/MLVXN/MicroLLM2:Q4_K_M
```
Unsloth Desktop
Docker Model Runner
How to use MLVXN/MicroLLM2 with Docker Model Runner:
```
docker model run hf.co/MLVXN/MicroLLM2:Q4_K_M
```

Lemonade

How to use MLVXN/MicroLLM2 with Lemonade:

Pull the model

# Download Lemonade from https://lemonade-server.ai/
lemonade pull MLVXN/MicroLLM2:Q4_K_M

Run and chat with the model

lemonade run user.MicroLLM2-Q4_K_M

List all available models

lemonade list

Atomic Chat

MLVXN commited on Aug 9

Commit

f0519ce

verified ·

1 Parent(s): 5797bb9

Update README.md

Browse files

Files changed (1) hide show

README.md +0 -7

README.md CHANGED Viewed

@@ -37,7 +37,6 @@ Plain GPT2 XL is a strong completer but not a chat model. MicroLLM2 adds:
 * ChatML format with `<|im_start|>` and `<|im_end|>` so conversations have clear user and assistant turns
 * Distilled instruction data from high quality teachers (GPT-4, GPT-3.5, Mixtral) plus identity reinforcement
-* Super optimized training for H100 so the full 1.5B could be tuned quickly without losing stability
 * Clean merge: no adapter needed at inference, just load like any GPT2 model
 No fancy claims here. It is still a 1.5B model with 1024 context. It will not beat 7B or larger models on broad knowledge, but it is far more useful than raw GPT2 XL for chatting, writing, and simple reasoning.
@@ -52,8 +51,6 @@ We wanted to keep it practical and affordable on a single H100 80GB.
 * **Identity:** 200 hand written identity prompts expanded to 10k during training so the model learns to answer consistently as MicroLLM2 by Maximalist Labs.
 * **Chat template:** `<|im_start|>user\n{prompt}<|im_end|>\n<|im_start|>assistant\n{response}<|im_end|>`
-Training was run outside the chat session on Lightning.ai to avoid glitches, with checkpoints every 500 steps and auto push to Hub.
 ## How to use
 ### Transformers
@@ -167,10 +164,6 @@ It will still admit it is based on GPT2 XL if you ask about its architecture, bu
 * `Modelfile` for Ollama
 * `microllm2-f16.gguf`, `microllm2-q8_0.gguf`, `microllm2-q4_k_m.gguf` GGUF weights
-## Reproduce and train
-The full training package is at `microllm2` on Lightning.ai. Key files are `train_distilled.py`, `config.py`, `identity.py`. It uses Hugging Face Transformers 4.44.2, PEFT 0.11.1, TRL 0.8.6, and expects an H100 or A100. See `deploy.sh` and `run_outside_chat.sh` for the one click run that keeps training alive outside the chat session.
 ## Credits
 Built by Maximalist Labs (MLVXN) on top of openai-community/gpt2-xl. Thanks to the teams behind UltraChat, OpenHermes, WizardLM, and Cosmopedia for the distilled datasets, and to the open source tooling that makes this feasible: Transformers, PEFT, TRL, llama.cpp, and Ollama.

 * ChatML format with `<|im_start|>` and `<|im_end|>` so conversations have clear user and assistant turns
 * Distilled instruction data from high quality teachers (GPT-4, GPT-3.5, Mixtral) plus identity reinforcement
 * Clean merge: no adapter needed at inference, just load like any GPT2 model
 No fancy claims here. It is still a 1.5B model with 1024 context. It will not beat 7B or larger models on broad knowledge, but it is far more useful than raw GPT2 XL for chatting, writing, and simple reasoning.
 * **Identity:** 200 hand written identity prompts expanded to 10k during training so the model learns to answer consistently as MicroLLM2 by Maximalist Labs.
 * **Chat template:** `<|im_start|>user\n{prompt}<|im_end|>\n<|im_start|>assistant\n{response}<|im_end|>`
 ## How to use
 ### Transformers
 * `Modelfile` for Ollama
 * `microllm2-f16.gguf`, `microllm2-q8_0.gguf`, `microllm2-q4_k_m.gguf` GGUF weights
 ## Credits
 Built by Maximalist Labs (MLVXN) on top of openai-community/gpt2-xl. Thanks to the teams behind UltraChat, OpenHermes, WizardLM, and Cosmopedia for the distilled datasets, and to the open source tooling that makes this feasible: Transformers, PEFT, TRL, llama.cpp, and Ollama.