How to use from
vLLM
Install from pip and serve model
# Install vLLM from pip:
pip install vllm
# Start the vLLM server:
vllm serve "theworker02/open-reason-medium"
# Call the server using curl (OpenAI-compatible API):
curl -X POST "http://localhost:8000/v1/completions" \
	-H "Content-Type: application/json" \
	--data '{
		"model": "theworker02/open-reason-medium",
		"prompt": "Once upon a time,",
		"max_tokens": 512,
		"temperature": 0.5
	}'
Use Docker
docker model run hf.co/theworker02/open-reason-medium
Quick Links

Open Reason medium (CPU)

A medium GPT-2-style causal LM trained from scratch on theworker02/open-reason pipeline v1.4.0. It is larger than theworker02/open-reason-small (~1.3M) and is not a 1B model. Do not confuse it with theworker02/open-reason-1b.

Parameters 13,867,008
Architecture GPT-2 scratch, n_layer=6, n_embd=384, n_head=6, vocab 8192, context 192
Steps 180
Batch size 2
Hardware Host CPU (torch 2.12.0+cpu). Docker was not installed. AMD GPU was not used. CUDA: false
Dataset theworker02/open-reason v1.4.0, 3175 SFT rows (all split)
Final loss 4.416
License Apache-2.0
Reddit Never used as a source
from transformers import AutoModelForCausalLM, AutoTokenizer

tok = AutoTokenizer.from_pretrained("theworker02/open-reason-medium")
model = AutoModelForCausalLM.from_pretrained("theworker02/open-reason-medium")

Companion small model: theworker02/open-reason-small. Dataset: theworker02/open-reason. Code: theworker02/open-reason.

Downloads last month
164
Safetensors
Model size
13.9M params
Tensor type
F32
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Dataset used to train theworker02/open-reason-medium