Hemlock-Jan-code-4B / README.md
nbeerbower's picture
Add model card with training configuration
2598226 verified
metadata
library_name: transformers
tags:
  - merlina
  - grimoire
  - text-generation
  - sft
datasets:
  - hemlang/Hemlock-SFT
base_model:
  - janhq/Jan-code-4b

Hemlock-Jan-code-4B

Training Configuration

Parameter Value
Training Mode SFT
Base Model janhq/Jan-code-4b
Learning Rate 9e-05
Epochs 1
Batch Size 1
Gradient Accumulation 32
Effective Batch Size 32
Max Sequence Length 2048
Optimizer paged_adamw_8bit
LR Scheduler cosine
Warmup Ratio 0.05
Weight Decay 0.01
Max Grad Norm 0.25
Seed 42
LoRA Rank (r) 128
LoRA Alpha 64
LoRA Dropout 0.05
Target Modules up_proj, down_proj, gate_proj, k_proj, q_proj, v_proj, o_proj
Quantization 4-bit (NF4)
GPU NVIDIA RTX A6000

Trained with Merlina

Merlina on GitHub