Model Details
- Base model:
skt/A.X-3.1-Light
- Architecture:
LlamaForCausalLM (unchanged from the base model)
- Parameters: approximately 7B
- Weight format: BF16
safetensors
- Chat template: the official A.X chat template
- Custom code: none
- Merge method: PEFT LoRA
merge_and_unload
- Base revision used for training and merge:
9b41bb2406472634d8812c0b8931fa40fa9a6c3a
Training Data
The SFT view contained 36,000 training examples and 3,000 validation
examples. Training and validation were kept separate at the source level.
The following AI Hub data domains were used:
- Civil law LLM pretraining and instruction-tuning data
- Criminal law LLM pretraining and instruction-tuning data
- Administrative law LLM pretraining and instruction-tuning data
- Corporate accounting standards data
- Essential medical knowledge data
- News article machine-reading data
- CoT-Fabric technology valuation data
Public evaluation benchmark data, including KMMLU-Pro, CLIcK, HLE,
SNU Ko-MuSR, Com2-main, and Original MuSR, was not used for SFT.
Training Procedure
- Objective: standard assistant-token causal language-model cross entropy
- Epochs: 1
- Learning rate:
5e-5
- Scheduler: cosine
- Warmup ratio:
0.03
- Weight decay:
0.01
- LoRA rank:
16
- LoRA alpha:
32
- LoRA dropout:
0.05
- LoRA target modules:
q_proj, k_proj, ,
The final training view used 800 reasoning examples. Their output tokens
accounted for approximately 24.37% of the total assistant target tokens.
The training format validator passed all 36,000 training and 3,000
validation examples.
import torch
from transformers import AutoModelForCausalLM, AutoTokenizer
model_id = "youngseok12/AX-3.1-Light-sft_v3_0"
tokenizer = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForCausalLM.from_pretrained(
model_id,
dtype=torch.bfloat16,
device_map="auto",
)
messages = [{"role": "user", "content": "대한민국의 수도는 어디인가요?"}]
inputs = tokenizer.apply_chat_template(
messages,
add_generation_prompt=True,
tokenize=True,
return_tensors="pt",
).to(model.device)
with torch.inference_mode():
outputs = model.generate(**inputs, max_new_tokens=64, do_sample=False)
print(tokenizer.decode(outputs[0][inputs["input_ids"].shape[-1]:], skip_special_tokens=True))
Usage with vLLM
vllm serve youngseok12/AX-3.1-Light-sft_v3_0 \
--dtype bfloat16 \
--max-model-len 32768
The repository is intended to be publicly accessible and directly loadable
by a standard vLLM OpenAI-compatible server. The model does not require
trust_remote_code=True.
Intended Use and Limitations
This model is intended for Korean-language research, controlled evaluation,
and experimentation with instruction-following and domain-oriented SFT.
It is not a substitute for professional legal, accounting, or medical
advice. Responses may contain factual errors, unsupported reasoning, or
format deviations. The reported training configuration does not imply an
official K-AI leaderboard result.
The AI Hub source data remains subject to its original access terms and
licenses. This repository contains model weights and documentation, not the
source training datasets.
License
The base model is distributed under the Apache License 2.0. Please review
the base model license and the terms applicable to the source training data
before using or redistributing this derivative model.