AutodevvLM v0.1

AutodevvLM v0.1 is a ~10.06M parameter decoder-only Transformer trained from scratch on CPU. It follows GPT-2 conventions (fused QKV attention, learned position embeddings, pre-LN blocks) and is small enough to run comfortably on a laptop.

It ships with both a PyTorch/Hugging Face checkpoint (model.safetensors, load with trust_remote_code=True) and GGUF builds for llama.cpp.

Model details

Property Value
Parameters 10,064,896
Architecture Decoder-only Transformer (GPT-2 style)
Layers 10
Model dim (n_embd) 256
Attention heads 8
FFN inner dim 1024
Vocab size 4096
Context (block_size) 256
Params dtype float32 (safetensors)

Files

File Description
config.json Model config (+ auto_map for trust_remote_code)
configuration_autodevlm.py AutoDevLMTinyGPTConfig (remote code)
modeling_autodevlm.py AutoDevLMTinyGPTForCausalLM (remote code)
model.safetensors Weights (float32, ~38 MB)
tokenizer.json Byte-level BPE tokenizer
tokenizer_config.json Tokenizer special tokens
autodevv-f16.gguf GGUF, F16 (~19.3 MB)
autodevv-q4_k_m.gguf GGUF, Q4_K_M (~6.5 MB)

Usage (PyTorch / transformers)

from transformers import AutoModelForCausalLM, AutoTokenizer
import torch

model = AutoModelForCausalLM.from_pretrained("<repo-id>", trust_remote_code=True)
tok = AutoTokenizer.from_pretrained("<repo-id>")

ids = tok("the tiny transformer", return_tensors="pt").input_ids
out = model.generate_text(ids, max_new_tokens=30, temperature=0.8, seed=0)
print(tok.decode(out[0]))

Usage (llama.cpp)

llama-cli -m autodevv-q4_k_m.gguf -p "the tiny transformer" -n 30 --temp 0.8 -s 0

Training

  • Trained from scratch on CPU (src/train.py, AdamW).
  • 50 steps: train loss 1.74 โ†’ 0.05, val loss 2.37 โ†’ 1.24.
  • Byte-level tokenizer (ByteLevel(use_regex=False)), vocab padded to 4096.

Intended use & limitations

This is a tiny educational model trained on a very small corpus. Outputs are simple and domain-limited; it is not a general-purpose assistant and should be used for learning and experimentation only.

License

MIT.

Downloads last month
-
Safetensors
Model size
10.1M params
Tensor type
F32
ยท
Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐Ÿ™‹ Ask for provider support