Release ULTRON-Core-512M-Instruct (SFT Aligned with <think> CoT)
Browse files- README.md +24 -0
- __init__.py +0 -0
- config.json +22 -0
- generation_config.json +9 -0
- model.safetensors +3 -0
- ultron_core_vocab.json +0 -0
README.md
ADDED
|
@@ -0,0 +1,24 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
language:
|
| 3 |
+
- en
|
| 4 |
+
- hi
|
| 5 |
+
- multilingual
|
| 6 |
+
license: apache-2.0
|
| 7 |
+
tags:
|
| 8 |
+
- instruct
|
| 9 |
+
- sft
|
| 10 |
+
- chain-of-thought
|
| 11 |
+
- pytorch
|
| 12 |
+
- from-scratch
|
| 13 |
+
- ultron-core
|
| 14 |
+
---
|
| 15 |
+
|
| 16 |
+
# ULTRON-Core-512M-Instruct (Generation 3 Aligned)
|
| 17 |
+
|
| 18 |
+
**ULTRON-Core-512M-Instruct** is an aligned conversational and reasoning foundation model trained from scratch with custom Multi-Head Latent Attention (MLA) and Chain-of-Thought (`<think>`) supervision.
|
| 19 |
+
|
| 20 |
+
## Architecture & Alignment
|
| 21 |
+
* **Base Parameters:** 562,382,848 (~512.05M Non-Embedding)
|
| 22 |
+
* **Attention:** Multi-Head Latent Attention (MLA: $d_c=512, d_R=64$) + QK-LayerNorm
|
| 23 |
+
* **Vocabulary:** 24,576 (Pure-Python Byte-Level BPE, NFC Standard)
|
| 24 |
+
* **Alignment:** Supervised Fine-Tuning (SFT) with dynamic prompt masking and multi-pattern `<think>` reasoning traces.
|
__init__.py
ADDED
|
File without changes
|
config.json
ADDED
|
@@ -0,0 +1,22 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"architectures": [
|
| 3 |
+
"UltronCoreForCausalLM"
|
| 4 |
+
],
|
| 5 |
+
"model_type": "ultron_core_instruct",
|
| 6 |
+
"vocab_size": 24576,
|
| 7 |
+
"hidden_size": 1024,
|
| 8 |
+
"intermediate_size": 3072,
|
| 9 |
+
"num_hidden_layers": 36,
|
| 10 |
+
"num_attention_heads": 16,
|
| 11 |
+
"head_dim": 64,
|
| 12 |
+
"mla_latent_dim": 512,
|
| 13 |
+
"mla_rope_dim": 64,
|
| 14 |
+
"max_position_embeddings": 2048,
|
| 15 |
+
"rms_norm_eps": 1e-06,
|
| 16 |
+
"rope_theta": 500000.0,
|
| 17 |
+
"initializer_range": 0.02,
|
| 18 |
+
"tie_word_embeddings": false,
|
| 19 |
+
"torch_dtype": "float16",
|
| 20 |
+
"total_parameters": 562382848,
|
| 21 |
+
"alignment_type": "SFT_ChainOfThought"
|
| 22 |
+
}
|
generation_config.json
ADDED
|
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"bos_token_id": 1,
|
| 3 |
+
"eos_token_id": 2,
|
| 4 |
+
"pad_token_id": 3,
|
| 5 |
+
"temperature": 0.7,
|
| 6 |
+
"top_p": 0.9,
|
| 7 |
+
"top_k": 40,
|
| 8 |
+
"repetition_penalty": 1.15
|
| 9 |
+
}
|
model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:87647ad069d2ec230988815aa134d8781720a5dee002c708be13210459210bd6
|
| 3 |
+
size 1124819904
|
ultron_core_vocab.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|