SoCode-v1-2B / README.md
theprint's picture
Upload README.md with huggingface_hub
2ba7e02 verified
|
Raw
History Blame Contribute Delete
1.49 kB
metadata
base_model: unsloth/Qwen3.5-2B
datasets:
  - OpceanAI/sota-coding
tags:
  - fine-tuned
  - lora
  - sft
  - auto-sft
language:
  - en
library_name: transformers

SoCode-v1-2B

A fine-tuned version of unsloth/Qwen3.5-2B trained on OpceanAI sota coding data using Auto-SFT — an automated hyperparameter search and supervised fine-tuning pipeline.

The base model was adapted to follow the style and content of the OpceanAI sota coding dataset. Expect improved performance on tasks similar to those represented in the training data.

Model Details

Property Value
Base model unsloth/Qwen3.5-2B
Training data OpceanAI/sota-coding
Fine-tuning epochs 1
Fine-tuning date 2026-07-21
Fine-tuning method LoRA (merged to full 16-bit)

Training Hyperparameters

LoRA

Parameter Value
r 64
alpha 256
dropout 0.07
target_modules ['q_proj', 'v_proj']

Training

Parameter Value
learning_rate 0.0002
batch_size 1
gradient_accumulation_steps 8
warmup_ratio 0.1
max_seq_length 512
quantization none

Usage

from transformers import AutoModelForCausalLM, AutoTokenizer

model     = AutoModelForCausalLM.from_pretrained("theprint/SoCode-v1-2B")
tokenizer = AutoTokenizer.from_pretrained("theprint/SoCode-v1-2B")

Generated by Auto-SFT