FINAL-Bench / lastbrain

🧠 lastbrain — Darwin V8

Darwin V8 기반 Claude Opus 증류 모델 (2B 파라미터)

• 👨 Father (Base) : Qwen/Qwen3.5-2B
• 👩 Mother (LoRA Adapter) : FINAL-Bench/Qwen3.5-2B-Opus-Distill-v1
• 👶 Child (This model) : FINAL-Bench/lastbrain — merged full-weight standalone

📦 특징

• Base : Qwen3.5-2B (2.3B 파라미터, 하이브리드 어텐션)
• Training : SFT + LoRA ( all-linear , rank=16, α=32)
• Teachers : Claude Opus 4.5 / 4.6, Claude Sonnet 4.6 (pre-generated reasoning traces)
• Data : 4,451 고품질 추론 궤적 (4개 공개 데이터셋)
• Merged : LoRA 어댑터가 base 가중치에 완전 통합되어 독립 실행 가능

🚀 빠른 사용법

from transformers import AutoTokenizer, AutoModelForCausalLM
import torch

model_id = "FINAL-Bench/lastbrain"
tok = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
model = AutoModelForCausalLM.from_pretrained(
    model_id, torch_dtype=torch.bfloat16, device_map="auto", trust_remote_code=True
)

messages = [
    {"role": "user", "content": "If a train travels 60 km in 45 minutes, what is its speed in km/h?"}
]
prompt = tok.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
inputs = tok(prompt, return_tensors="pt").to(model.device)

with torch.no_grad():
    outputs = model.generate(
        **inputs,
        max_new_tokens=800,
        do_sample=False,
        pad_token_id=tok.eos_token_id,
    )
print(tok.decode(outputs[0][inputs.input_ids.shape[1]:], skip_special_tokens=True))

예시 출력 :

To find the speed of the train in km/h, we need to convert the given time from minutes to hours.

**Given:**
- Distance = 60 km
- Time = 45 minutes

**Step 1: Convert time to hours**
Since there are 60 minutes in 1 hour:
$\text{Time in hours}=\frac{45}{60}=0.75\text{ hours}\backslash text\{Time\; in\; hours\}\; =\; \backslash frac\{45\}\{60\}\; =\; 0.75\; \backslash text\{\; hours\}$Time in hours=6045​=0.75 hours

**Step 2: Calculate speed**
$\text{Speed}=\frac{60}{0.75}=80\text{ km/h}\backslash text\{Speed\}\; =\; \backslash frac\{60\}\{0.75\}\; =\; 80\; \backslash text\{\; km/h\}$Speed=0.7560​=80 km/h

**Final Answer:** The speed of the train is **80 km/h**.

🧬 Darwin V8 학습 파이프라인

[Qwen/Qwen3.5-2B] ──── Base 모델 (동결)
        +
[4,451 Claude Opus/Sonnet reasoning traces]
        ↓
[SFT Training]
  - LoRA (all-linear, r=16, α=32)
  - Learning rate: 2e-4 (V8 rule: ×10 FullFT)
  - 2 epochs, bf16, 8×B200 DDP
  - Loss: 1.33 → 1.10 (-17%)
  - Token accuracy: 68% → 72% (+4%p)
        ↓
[LoRA merge into base weights]
        ↓
[lastbrain] ← 이 모델

📊 학습 데이터 구성

🎯 설계 철학 (Darwin V8)

1. LoRA Without Regret — all-linear target, high LR, 작은 rank도 OK
2. Response Distillation — pre-generated Opus traces로 비용 효율적 증류
3. Merge-and-Deploy — LoRA 어댑터 통합 후 추가 의존성 없이 배포

🔁 재현 방법

이 모델은 다음 두 컴포넌트를 merge하여 만들어졌습니다:

from transformers import AutoModelForCausalLM
from peft import PeftModel
import torch

base = AutoModelForCausalLM.from_pretrained(
    "Qwen/Qwen3.5-2B", torch_dtype=torch.bfloat16
)
model = PeftModel.from_pretrained(
    base, "FINAL-Bench/Qwen3.5-2B-Opus-Distill-v1"
)
merged = model.merge_and_unload()
merged.save_pretrained("./lastbrain")

📝 샘플 테스트 결과 (4문제)

Markdown/LaTeX/Step-by-Step 구조화된 답변 자연스럽게 생성

⚠️ 제한 사항

• 규모 : 2.3B 파라미터 (소형 모델)
• 한국어 계산 정확성 : 때로 숫자 오류 발생 가능 (소형 모델 한계)
• 긴 컨텍스트 : 학습 시 max_length=4,096으로 학습됨
• <think> 태그 : 명시적 사용 낮음 (reasoning을 본문에 통합)

🪪 라이선스

• Base model: Apache 2.0 (Qwen)
• 학습 데이터: 각 데이터셋 개별 라이선스 참조
• 이 모델: Apache 2.0

🙏 크레딧

• Base : Qwen team (Alibaba)
• Teacher : Anthropic (Claude Opus 4.5/4.6, Sonnet 4.6)
• 데이터 공개 : nohurry, TeichAI
• Training & Release : FINAL-Bench / VIDRAFT_LAB

🔗 관련 모델

• 🧠 FINAL-Bench/Qwen3.5-2B-Opus-Distill-v1 — 이 모델의 LoRA 어댑터 단독 버전
• ⚡ FINAL-Bench/Qwen3.5-2B-Opus-SDPO-v1 — Phase 4 SDPO 자기증류 강화본

Darwin V8 · Part of the evolutionary model merging series by VIDRAFT_LAB