🏯 Hanok LLM 1.0

Korean-first modular language model stack

🌐 Languages: English Β· ν•œκ΅­μ–΄

A native PyTorch Transformer stack for Korean-language model research, training, inference, checkpointing, and data workflows.

Python 3.10+ PyTorch Transformers Datasets Windows 11 CUDA License

Hanok LLM overview

33 Layers Β· 1,280 Hidden Β· 10 Heads Β· SwiGLU 5,881 Β· 1B Parameters

Repository: https://github.com/seoan1024/Hanok-LLM.git
Research base: https://github.com/seoan1024/Korean-llm.git


✦ What is Hanok?

Hanok LLM is built on the foundation of continuous research from the Korean-llm project.
It is a modular Korean language model stack for research and experimentation.

The entire system lives under a maintainable Python package (src/hanok/) with strict separation of concerns. Every major subsystem β€” model, data, training, inference, evaluation, export and monitoring β€” can be inspected, extended or replaced independently.

Layer Responsibility
Model Native decoder-only Transformer (KoreanLLM) with RMSNorm, RoPE, SwiGLU, tied embeddings and KV cache
Data Hugging Face dataset loading, Parquet caching, Korean-oriented field parsing and response-only SFT masking
Training Pretrain β†’ SFT orchestration, gradient accumulation, cosine schedule, checkpoint resume, optional GUI
Inference Native .pth loading + temperature / top-k / top-p / repetition-penalty generation
Evaluation Perplexity and a reproducible Korean multiple-choice regression benchmark
Export Standard Transformers Llama bundle, llama.cpp GGUF, and Ollama model creation
UI Optional real-time training monitor GUI

Build a Korean LLM stack you can inspect and experiment with. Native PyTorch. Explicit defaults. Local checkpoints. Practical Windows + CUDA workflow.

Trained model weights are not included. Export commands convert a checkpoint supplied by the user.


🧠 Architecture in depth

Hanok architecture

Core configuration

Parameter Value Notes
Transformer blocks 33 n_layers
Hidden dimension 1,280 dim
Attention heads 10 n_heads
Head dimension 128 dim // n_heads
SwiGLU intermediate 5,881 intermediate_size
Total parameters 1,000,000,000 tied embeddings, vocab 30,004
Normalization RMSNorm eps=1e-6
Positional encoding RoPE ΞΈ = 10,000
Residual style Pre-norm Attention & FFN
Embedding / output Tied weights output.weight = embed.weight
KV cache Supported Per-layer, detached
Default training seq length 1,024 TrainingConfig.max_seq_len; configurable
Direct model constructor 4,096 KoreanLLM(max_seq_len=4096) default RoPE table length
Training / CLI context 1,024 Training, inference, evaluation, and export defaults
Model vocabulary 30,004 Model vocabulary size; tokenizer vocabulary must not exceed this
Total parameter count 1,000,000,000 (1B) 1B decoder-only Transformer with tied embeddings

The sequence limit is the RoPE table length allocated for that model instance, not a fixed architecture-wide ceiling. The KoreanLLM constructor defaults to 4,096 tokens, while the training and CLI loading paths default to 1,024; pass a larger --max-seq-len to configure a longer context. Generation counts the prompt and new tokens together, truncates an overlong prompt from the left, and stops at the configured context limit. Context lengths above 1,024 are configurable but are not the default training length and have not been validated for quality here; memory use also grows with sequence length.

Building blocks

RMSNorm
Root-mean-square normalization without mean centering. Lightweight and stable for deep residual stacks.

RoPE (Rotary Positional Embeddings)
Applied to both queries and keys. Precomputed cosine/sine tables are registered as buffers and sliced by absolute position (supports KV-cache incremental decoding).

Attention

  • Multi-head self-attention with F.scaled_dot_product_attention
  • Causal masking by default (is_causal=True when no external mask is given)
  • Optional external attention mask
  • KV cache concatenation along the sequence dimension for efficient generation

SwiGLU Feed-Forward
Three linear projections (w1, w2, w3) with no bias:
w2(SiLU(w1(x)) * w3(x)). Intermediate size is 5,881. Total parameter count is 1,000,000,000 (1B) with tied input/output embeddings and 30,004 vocabulary size.

TransformerBlock
Pre-norm residual:
x = x + Attention(RMSNorm(x))
x = x + SwiGLU(RMSNorm(x))

KoreanLLM

  • Embedding β†’ 33 Γ— TransformerBlock β†’ final RMSNorm β†’ tied output projection
  • Weight initialization: normal(std=0.02) for Linear and Embedding
  • Training path uses torch.utils.checkpoint (non-reentrant) for memory efficiency
  • Loss: token-level cross-entropy with ignore_index=-100, reduced by valid target count (handles empty-target batches safely)

Design philosophy

The architecture deliberately stays close to modern decoder-only practices (RMSNorm + RoPE + SwiGLU + tied embeddings) while remaining fully transparent. There are no opaque custom CUDA kernels or closed-source components β€” every tensor operation is visible in pure PyTorch.


⚑ Training stack

Training pipeline

Default TrainingConfig

stage                      auto | pretrain | sft
batch_size                 4
accumulation_steps         8          # effective batch = 32
learning_rate              3e-4
sft_learning_rate          2e-5
warmup_steps               1,000
scheduler                  Cosine
weight_decay               0.1
max_steps                  100,000    # manual-stage default
pretrain_steps             100,000
sft_steps                  10,000
eval_interval              5,000
max_seq_len                1024
num_workers                4
use_bfloat16               True
use_8bit_optimizer         True       # AdamW8bit β†’ AdamW fallback
validation_split           0.02
validation_batches         32
enable_gui                 True
seed                       42
checkpoint_dir             checkpoints
force_redownload           False
stream_datasets            False
samples_per_dataset        None
resume_from_checkpoint     None

Training stages

                    β”Œβ”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”
                    β”‚   DATA SOURCES   β”‚
                    β””β”€β”€β”€β”€β”€β”€β”€β”€β”¬β”€β”€β”€β”€β”€β”€β”€β”€β”€β”˜
                             β”‚
                    β”Œβ”€β”€β”€β”€β”€β”€β”€β”€β–Όβ”€β”€β”€β”€β”€β”€β”€β”€β”€β”
                    β”‚ CACHE / PARQUET  β”‚
                    β””β”€β”€β”€β”€β”€β”€β”€β”€β”¬β”€β”€β”€β”€β”€β”€β”€β”€β”€β”˜
                             β”‚
                 β”Œβ”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β–Όβ”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”
                 β”‚       PRETRAIN        β”‚
                 β”‚ Web + Wiki + GSM8K-Ko 믹슀 β”‚
                 β”‚  (100k steps default) β”‚
                 β””β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”¬β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”˜
                             β”‚ checkpoint
                 β”Œβ”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β–Όβ”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”
                 β”‚          SFT          β”‚
                 β”‚ BCCard + KoAlpaca + AI Hub + Hanok β”‚
                 β”‚ (10k steps default)   β”‚
                 β”‚ response-only labels  β”‚
                 β””β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”¬β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”˜
                             β”‚
                    β”Œβ”€β”€β”€β”€β”€β”€β”€β”€β–Όβ”€β”€β”€β”€β”€β”€β”€β”€β”
                    β”‚  NATIVE .PTH    β”‚
                    β””β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”€β”˜
  • auto mode: runs full pretraining first, then starts SFT from the latest pretrain checkpoint (fresh optimizer & schedule for SFT).
  • Checkpoint resume: supported for both stages.
  • SFT label masking: only response tokens contribute to the loss; prompt tokens are set to -100.
  • Pretraining packing: complete documents are joined into full 1,024-token blocks; Parquet rows are shuffled within row groups to reduce repeated reads.
  • SFT updates: rank-64 LoRA adapters (alpha 128) train while the pretrained base weights stay frozen. Native checkpoints retain the adapters; HF export merges them into the base layers.
  • Validation: pretraining holds out about 2% by Parquet row group; SFT uses a 2% random split. Evaluated every eval_interval steps (up to validation_batches).
  • GUI: optional real-time monitor (enable_gui=True by default; disable with --no-gui). It separates Pretraining and SFT curves by color, includes Matplotlib zoom/pan controls, loads the latest checkpoint for CPU chat, carries the latest 12 turns into the next prompt, limits chat generation length, and can clear the chat view. The standalone infer command still generates from one supplied prompt per invocation.

Optimizer & scheduler

  • Primary: AdamW8bit (bitsandbytes) when available
  • Fallback: standard AdamW
  • Learning-rate schedule: Cosine annealing with linear warmup (1,000 steps)

Memory & precision

  • BF16 autocast on CUDA when use_bfloat16=True
  • Gradient checkpointing active during training
  • Pin memory for DataLoader on CUDA devices
  • SFT uses rank-64 LoRA adapters (alpha 128) while keeping the pretrained base frozen.

πŸ–₯️ Validated workstation

Windows 11 + RTX 5090 profile
Component Recommended profile
OS Windows 11
GPU RTX 5090 Laptop GPU (reference profile; 1B peak not measured)
VRAM 24 GB may be tight; measure before a full run
System RAM 24 GB or more
CPU High-performance multi-core
Python 3.10+ (documented on 3.11)
Precision BF16 on supported CUDA hardware
Storage Fast SSD (datasets + checkpoints grow quickly)

Memory note With sequence length 1,024, default batch size and accumulation, FP32 1B model weights with BF16 autocast, AdamW8bit, and gradient checkpointing, the repository does not publish a measured peak-VRAM result. Actual use depends on GPU, batch size, sequence length, optimizer, and CUDA allocation state; LoRA SFT has a different memory profile. Actual usage also depends on batch size, sequence length, optimizer state, whether gradient checkpointing is active, and whether you are training or only running inference. Longer sequences or larger effective batches will require more VRAM or reduced accumulation.


πŸ§ͺ Validation & roadmap status

Validation status
Claim Status
Validation environment RTX 5090 Laptop GPU
Model construction and native inference code Included
Training, checkpoint, and resume code Included
Automated test suite included No
Model Card & Data Card included βœ…
Research and experimentation focus βœ…
Trained weights and benchmark results Not included
Transformers Llama export code βœ… Included
llama.cpp GGUF / Ollama export code βœ… Included; external tools required
Official Korean quality benchmark results Not run

🧩 Technology stack

Technology stack
Category Libraries / tools
Core Python 3.10+, PyTorch 2.x
Tokenization Hugging Face Transformers (EleutherAI/polyglot-ko-1.3b)
Data Hugging Face Datasets, PyArrow, Pandas
Optimization bitsandbytes (optional 8-bit AdamW)
Monitoring matplotlib + optional GUI monitor
Packaging setuptools, pyproject.toml, CLI entrypoint hanok

πŸ“¦ Project structure

Hanok-LLM/
β”œβ”€β”€ src/hanok/
β”‚   β”œβ”€β”€ model/
β”‚   β”‚   └── architecture.py          # KoreanLLM (RMSNorm Β· RoPE Β· SwiGLU Β· KV cache)
β”‚   β”œβ”€β”€ data/
β”‚   β”‚   └── legacy.py                # DatasetManager, LocalKoreanDataset, collate, Parquet cache
β”‚   β”œβ”€β”€ training/
β”‚   β”‚   β”œβ”€β”€ config.py                # TrainingConfig dataclass (all defaults)
β”‚   β”‚   β”œβ”€β”€ trainer.py               # Main orchestration (auto / pretrain / sft)
β”‚   β”‚   β”œβ”€β”€ optimizer.py             # AdamW / AdamW8bit selection
β”‚   β”‚   β”œβ”€β”€ scheduler.py             # Cosine + warmup
β”‚   β”‚   β”œβ”€β”€ checkpoint.py            # Save / load / find_latest
β”‚   β”‚   └── reproducibility.py       # Seed & distributed helpers
β”‚   β”œβ”€β”€ inference/
β”‚   β”‚   β”œβ”€β”€ loader.py                # Checkpoint + tokenizer loading
β”‚   β”‚   └── generation.py            # Autoregressive generation with KV cache
β”‚   β”œβ”€β”€ evaluation/
β”‚   β”‚   β”œβ”€β”€ perplexity.py
β”‚   β”‚   β”œβ”€β”€ benchmark.py
β”‚   β”‚   └── benchmark_ko.jsonl       # local sanity suite, not an official score
β”‚   β”œβ”€β”€ export/
β”‚   β”‚   β”œβ”€β”€ huggingface.py           # Map Hanok weights to standard Transformers Llama
β”‚   β”‚   └── ollama.py                # llama.cpp GGUF conversion, Modelfile and Ollama creation
β”‚   β”œβ”€β”€ ui/
β”‚   β”‚   └── monitor.py               # Optional training GUI
β”‚   └── cli.py                       # `hanok` entrypoint (train / infer)
β”œβ”€β”€ assets/svg/                      # README visual kit
β”œβ”€β”€ MODEL_CARD.md
β”œβ”€β”€ DATA_CARD.md
β”œβ”€β”€ CONTRIBUTING.md
β”œβ”€β”€ SECURITY.md
β”œβ”€β”€ pyproject.toml
└── requirements.txt

πŸ› οΈ Installation and execution

macOS (Apple Silicon)

python3 --version                 # Python 3.10+
python3 -m venv .venv
source .venv/bin/activate
python -m pip install --upgrade pip
python -m pip install -e .
python -c "import torch; print('MPS:', torch.backends.mps.is_available())"
hanok --help

When MPS is available, training and inference select it automatically. Use --device mps for inference when you want to make the choice explicit.

Windows 11

# 1. Create / activate a Python 3.10+ environment
python --version

# 2. Install PyTorch with the CUDA build that matches your driver
#    β†’ https://pytorch.org (select CUDA version carefully)

# 3. Install the project in editable mode
python -m pip install -e .

# 4. Optional: 8-bit optimizer support
python -m pip install -e ".[8bit]"

# 5. Verify the CLI is available
hanok --help

Check CUDA before training:

python -c "import torch; print('CUDA:', torch.cuda.is_available()); print(torch.cuda.get_device_name(0) if torch.cuda.is_available() else 'CPU')"

Dependencies (from requirements.txt / pyproject.toml)

torch
transformers
datasets
safetensors
bitsandbytes          # optional, for 8-bit AdamW
pyarrow
pandas
requests
tqdm
matplotlib

🎯 Quick start

SFT (start from a pretrained checkpoint)

hanok train --stage sft --lora-rank 64 --resume-from-checkpoint checkpoints/pretrain/hanok_llm_100000.pth --max-steps 30000 --no-gui

The default SFT run combines BCCard/gemma-4-31B-korean-on-policy-150k, beomi/KoAlpaca-v1.1a, one pinned shard of AI Hub daily multi-session conversations, and the bundled Hanok identity/sanity examples. Local identity rows are repeated twice and the 29-row basic QA set four times. AI Hub terms and access conditions apply to the third-party mirror. SFT stops if any configured dataset fails to load. For multi-turn rows, preceding turns are masked context and the last user's first following assistant answer is trained. To select datasets explicitly, repeat --dataset (specifying it replaces the default SFT list):

hanok train --stage sft \
  --lora-rank 64 \
  --resume-from-checkpoint checkpoints/pretrain/hanok_llm_100000.pth \
  --max-steps 30000 \
  --dataset BCCard/gemma-4-31B-korean-on-policy-150k \
  --dataset beomi/KoAlpaca-v1.1a \
  --no-gui

Pretraining

hanok train --stage pretrain --no-gui

For a small configuration check, limit both the dataset and the number of steps:

hanok train --stage pretrain --dataset eliceai/korean-webtext-edu --samples-per-dataset 32 --max-steps 1 --no-gui

Auto pipeline (pretrain β†’ SFT)

hanok train --no-gui

The automatic path first completes pretraining (or resumes the latest pretrain checkpoint), then starts SFT from that checkpoint with a fresh optimizer and learning-rate schedule when no SFT checkpoint exists. If an SFT checkpoint is present, it resumes that SFT run. To start a clean training run, remove or rename existing checkpoints/pretrain and checkpoints/sft directories.

To run SFT directly from a pretraining checkpoint:

hanok train --stage sft --lora-rank 64 `
  --resume-from-checkpoint checkpoints/pretrain/hanok_llm_100000.pth `
  --max-steps 30000 --no-gui

Useful training flags

Flag Description
--stage {auto,pretrain,sft} Training stage
--dataset REPOSITORY[:CONFIG] Add an explicit dataset; repeat for multiple datasets
--no-gui Disable the optional training monitor
--stream-datasets Stream instead of full download
--samples-per-dataset N Limit examples per dataset (smoke / debug)
--max-steps N Override max training steps
--batch-size N Override batch size
--max-seq-len N Set the RoPE context and training sequence length (default: 1,024)
--lora-rank N SFT LoRA rank (default: 64)
--resume-from-checkpoint Path or latest
--force-redownload Re-download datasets

πŸ’¬ Inference

hanok infer `
  --checkpoint checkpoints/pretrain/hanok_llm_10000.pth `
  --prompt "μ•ˆλ…•ν•˜μ„Έμš”. λ„ˆλŠ” λˆ„κ΅¬λ‹ˆ?" `
  --max-tokens 128

On macOS, the equivalent multiline command is:

hanok infer \
  --checkpoint checkpoints/pretrain/hanok_llm_10000.pth \
  --prompt "μ•ˆλ…•ν•˜μ„Έμš”. λ„ˆλŠ” λˆ„κ΅¬λ‹ˆ?" \
  --max-tokens 128 \
  --device mps

Pretraining checkpoints can also generate text. They are trained for next-token continuation rather than instruction following, so use --prompt-format raw:

hanok infer \
  --checkpoint checkpoints/pretrain/hanok_llm_10000.pth \
  --prompt "ν•œμ˜₯은 ν•œκ΅­μ˜ 전톡 κ°€μ˜₯으둜" \
  --prompt-format raw \
  --max-tokens 128

Generation controls

Flag Default Description
--temperature 0.6 Softmax temperature (> 0 required)
--top-k 40 Keep only top-k logits
--top-p 0.95 Nucleus sampling
--repetition-penalty 1.3 Penalize already-generated tokens
--max-seq-len 1024 Context window used at load time
--device auto cuda / mps / cpu
--prompt-format qa qa for SFT-style prompts, raw for pretraining continuation
--greedy off Disable sampling and choose the highest-probability token each step

How generation works

  1. Prompt is wrapped in the instruction format:
    ### 질문: {prompt}\n### 응닡:
  2. Tokens are generated autoregressively with a KV cache.
  3. With sampling enabled: temperature scaling β†’ optional top-k filtering β†’ softmax β†’ optional top-p nucleus filtering β†’ multinomial sampling. With --greedy, the highest-probability token is selected directly.
  4. Repetition penalty is applied to previously seen tokens.
  5. Stops on EOS or when the sequence length safety limit is reached.
  6. Only the text after ### 응닡: is returned.

πŸ—‚οΈ Data sources & pipeline

Default datasets

Stage Dataset Config Split Notes
Pretraining eliceai/korean-webtext-edu Pinned revision c949654b Β· data-*-of-00018.arrow train Educationally filtered Korean web text
Pretraining wikimedia/wikipedia 20231101.ko train Korean Wikipedia dump
Pretraining kuotient/gsm8k-ko default train Korean translated GSM8K math dataset
SFT BCCard/gemma-4-31B-korean-on-policy-150k default train Instruction/output fields; quality depends on upstream generation
SFT beomi/KoAlpaca-v1.1a default train Korean instruction/output examples
SFT mkd-chanwoo/aihub_conversation Pinned revision c1f42ab Β· one JSONL shard train Everyday Korean multi-session conversations; AI Hub terms apply
SFT Local Hanok identity examples local train Name, role, capabilities, and honest limits; repeated 2 times
SFT Local Hanok sanity curriculum local train 29 basic QA examples, repeated 4 times; not independently fact-checked

Tokenizer: EleutherAI/polyglot-ko-1.3b
A dedicated pad token <|pad|> is added when the tokenizer does not already provide a distinct pad token.

Data handling details

  • Hugging Face downloads are cached under ./datasets/cache.
  • Downloaded rows are stored as Parquet and recorded in a manifest.
  • Field parsing supports multiple common schemas:
    text / content / document / body,
    instruction / input / output,
    question / response, question / answer, prompt / response, role/content conversation arrays, and ChatML.
  • SFT mode masks the prompt portion; only response tokens contribute to the loss. Multi-turn rows retain preceding turns as context and supervise the first assistant answer after the final user turn; other assistant turns in that row are not targets.
  • EOS is appended; PAD positions are ignored via ignore_index=-100.
  • Optional streaming and samples_per_dataset limits are available for rapid iteration.

Always verify the original dataset licenses and terms of use before training.
This repository does not redistribute or re-license third-party data.


πŸ“€ Checkpoint export & evaluation

Export a Hanok checkpoint to the standard Transformers LlamaForCausalLM format:

hanok export --checkpoint checkpoints/sft/hanok_llm_05000.pth --format hf --output exports/hanok-hf

# GGUF conversion (requires an existing llama.cpp checkout)
$llamaCppDir = Read-Host "Enter the absolute path to your llama.cpp checkout"
hanok export --checkpoint checkpoints/sft/hanok_llm_05000.pth --format gguf `
  --output exports/hanok-f16.gguf --llama-cpp-dir $llamaCppDir

# Quantize and register in Ollama (requires llama-quantize and Ollama)
hanok export --checkpoint checkpoints/sft/hanok_llm_05000.pth --format ollama `
  --output exports/hanok-f16.gguf --llama-cpp-dir $llamaCppDir `
  --quantization Q4_K_M --ollama-name hanok:latest

Load the HF folder with AutoModelForCausalLM.from_pretrained("exports/hanok-hf"). GGUF/Ollama export runs external llama.cpp and optional quantization tools. Ollama has no deploy command: the export command registers the model with ollama create; run it with ollama run hanok:latest.

If you create the Modelfile without the combined export command, use:

ollama create hanok:latest -f exports/hanok-f16.Modelfile
ollama run hanok:latest

Run hanok evaluate --checkpoint ... to evaluate the bundled 15-question Korean multiple-choice suite and three Korean free-response prompts with deterministic generation. The JSON report includes accuracy, answer format rate, Hangul ratio, repeated-token ratio, empty-response rate, choice-letter perplexity, and every generated response. Choice-letter perplexity only describes this small test format and is not comparable to training validation loss. This is a regression check, not an official or comprehensive Korean capability benchmark.

This repository does not include trained weights. Hanok initializes from scratch; export does not add pretrained Llama knowledge or improve a checkpoint's quality. Results depend on the training data and training budget.


πŸ”¬ Model Card & Data Card

Document Purpose
MODEL_CARD.md Architecture summary, intended use, limitations, tokenizer notes
DATA_CARD.md Dataset sources, preprocessing behavior, reproducibility caveats
CONTRIBUTING.md How to contribute
SECURITY.md Security reporting process

🧭 Design principles

  1. Inspectability β€” every layer is ordinary PyTorch; no black-box kernels.
  2. Explicit defaults β€” training hyperparameters are declared in one place (TrainingConfig).
  3. Local-first β€” checkpoints are native .pth files you can load without a remote registry.
  4. Windows + CUDA practicality β€” the documented path is a real workstation workflow, not a theoretical Linux-only lab.
  5. Clear scope β€” this repository provides implementation code; trained weights and benchmark results are not included.

πŸ“œ License

See LICENSE for the repository license.
Third-party datasets, tokenizers and dependencies carry their own terms. Always review them before any commercial use.


🏯 HANOK · KOREAN LLM ENGINEERING

Inspectable architecture Β· Practical training Β· Native inference Β· Windows/CUDA workflow Β· Research and experimentation

Repo: Hanok-LLM Β· Research base: Korean-llm

Hanok closing panel
Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. πŸ™‹ Ask for provider support

Datasets used to train seoan1024/Hanok-LLM