--- license: apache-2.0 language: - en tags: - ternary - quantized - atlas - tq1 - cpu-optimized - trilm-subln - llm - cpu-llm - edge-ai - no-gpu - efficient-inference base_model: SpectraSuite/TriLM_1.5B_Unpacked pipeline_tag: text-generation library_name: atlas --- # TriLM-1.5B-ATLAS This repository contains a highly optimized **TQ1 quantized version** of the official `SpectraSuite/TriLM_1.5B_Unpacked` model for the **ATLAS Engine** ecosystem, designed for native, ultra-low-latency CPU inference without any GPU requirement. > Packed using the unified `pack_to_atlas.py` toolchain (v2.10.0) with BF16 weight scale correction. --- ## Engine Specifications | Property | Value | |---|---| | **Format** | ATLAS Binary (`.atlas`), format_version=2 | | **Quantization** | TQ1.0 — Ternary Weight Packing (Base-3, ~1.58 bits/weight) | | **Target** | Native CPU — Intel AVX2 (Haswell 2013+), no GPU needed | | **File Size** | 0.65 GB | | **Inference Speed** | ~20 tok/s (f32 bypass) | | **Description** | 24 layers, 2048 hidden, 6144 intermediate | ### Architecture | Component | Detail | |---|---| | Base Model | `SpectraSuite/TriLM_1.5B_Unpacked` | | Architecture | trilm-subln | | Layers | 24 | | Hidden Size | 2048 | | Intermediate Size | 6144 | | Attention Heads | 32 (GQA, 32 KV heads) | | Head Dim | 64 | | RoPE Theta | 10000 | | Vocabulary | 50432 | | Context Window | 4096 | > **f32 bypass active:** hidden=2048 ≤ 2048, auto-enabled ### Verification During pre-release evaluation, this quantized derivative demonstrated correct convergence: - **T=0 (argmax):** `"The capital of France is Paris."` — correct deterministic output - **T=0.7 (sampling):** Coherent structured generation with sensible continuation > *Note on scale mathematics:* the legacy dequantization path divides by the scale factor rather than multiplying. Since this is a constant across all logits for any given output row, the relative probability distribution remains identical under softmax normalization — no effect on output quality. --- ## Prompt Template This is a **base model** — no chat template. Use raw text continuation: ``` The capital of France is ``` --- ## Usage ### Python ```bash git clone https://github.com/xxxn3m3s1sxxx/ATLAS-TQ1_0.git ``` ```python from atlas_infer import AtlasModel model = AtlasModel("TriLM-1.5B-ATLAS.tq1.atlas") output = model.generate_c( "The capital of France is", max_new_tokens=50, temperature=0.7, top_k=40, ) print(output) ``` ### C++ CLI (standalone, no Python required) ```bash atlas --model TriLM-1.5B-ATLAS.tq1.atlas --prompt "The capital of France is" --max-tokens 50 ``` ### SSE Web Server ```bash python atlas_server.py --model TriLM-1.5B-ATLAS.tq1.atlas --port 8080 curl http://localhost:8080/v1/chat/completions \ -H "Content-Type: application/json" \ -d '{"prompt": "The capital of France is", "max_tokens": 50}' ``` --- ## What is ATLAS? **ATLAS** is a CPU inference engine for BitNet b1.58 ternary-quantized models. It repacks HuggingFace safetensors into the **TQ1.0 format** (5 ternary trits per byte, Base-3 encoding, ~1.58 bits/weight) and runs fast inference via a C++ DLL + Python wrapper. | Feature | Description | |---|---| | **No GPU required** | Runs on any x86-64 CPU with AVX2 (Intel Haswell 2013+, AMD Excavator 2015+) | | **Hybrid matmul** | FFN tensors in int8, QKV/O in TQ1-packed, per-tensor dispatch | | **int4 FFN mode** | Halves FFN memory bandwidth for 18-26% speedup (7B/10B) | | **f32 bypass** | Auto-enabled for small models (≤1B) and SubLN architectures | | **Ring buffer KV cache** | Extended context via NTK-aware RoPE scaling | | **Standalone C++ CLI** | No Python or PyTorch required at runtime | | **SSE web server** | FastAPI-based `/v1/chat/completions` with prompt caching | ### Links - **Engine source code**: [github.com/xxxn3m3s1sxxx/ATLAS-TQ1_0](https://github.com/xxxn3m3s1sxxx/ATLAS-TQ1_0) --- ## License | Component | License | |-----------|---------| | Base Model (SpectraSuite/TriLM_1.5B_Unpacked) | [Apache 2.0](https://www.apache.org/licenses/LICENSE-2.0) | | ATLAS Engine | [Apache 2.0](https://github.com/xxxn3m3s1sxxx/ATLAS-TQ1_0) | | This Quantized Derivative | [Apache 2.0](https://www.apache.org/licenses/LICENSE-2.0) |