Vertex 0.6 100M β€” 1024-ctx Base

Base (text-completion) model of the Vertex 0.6 family by Vertex Research, pretrained from scratch on 10.01B tokens at 1024 context. This is the raw pretrain checkpoint: no instruction tuning, no chat alignment. Instruct and reasoning variants (with extended context) are released separately.

Model details

Parameters 96.75M (tied embeddings)
Architecture Qwen3-based transformer
Hidden size / layers 576 / 16
Attention 9 heads, 3 KV heads (GQA), head_dim 64
FFN size 2304 (SwiGLU)
Context length 1024
RoPE theta 10,000
Vocab 32,768 (Supra2-100M tokenizer, digit-split)
Precision trained fp32 compute-graph via torch.compile

Training

  • Data: ~10.01B tokens β€” English web text from Ultra-FineWeb plus a code slice, packed into 1024-token blocks, seeded shuffle, single epoch.
  • Schedule: 101,819 optimizer steps, global batch 98,304 tokens (12 Γ— grad-accum 8 Γ— 1024), LR 3e-3 cosine β†’ 3e-4, 300 warmup steps.
  • Hardware: single RTX 4060 Ti 16GB, ~35K tokens/s, ~3.5 days wall clock.

Usage

from transformers import AutoTokenizer, AutoModelForCausalLM

tok = AutoTokenizer.from_pretrained("VertexResearch/Vertex-0.6-100M-1024-ctx-Base")
model = AutoModelForCausalLM.from_pretrained("VertexResearch/Vertex-0.6-100M-1024-ctx-Base")

ids = tok("The capital of France is", return_tensors="pt").input_ids
out = model.generate(ids, max_new_tokens=30)
print(tok.decode(out[0]))

This is a completion model β€” prompt it with text to continue, not with chat messages.

Limitations

These models are not the most coherent yet and need more tuning: expect rambling, repetition, and inconsistent answers, especially over longer generations.

97M parameters and 10B tokens: expect fluent English and basic factual/code completion, frequent factual errors, and repetition under greedy decoding. Knowledge cutoff of the web corpus is approximately April 2024.

Downloads last month
161
Safetensors
Model size
96.8M params
Tensor type
F32
Β·
Inference Providers NEW
This model isn't deployed by any Inference Provider. πŸ™‹ Ask for provider support

Dataset used to train VertexResearch/Vertex-0.6-100M-1024-ctx-Base

Collection including VertexResearch/Vertex-0.6-100M-1024-ctx-Base