SoCode-v1-2B / README.md
theprint's picture
Upload README.md with huggingface_hub
2ba7e02 verified
|
Raw
History Blame Contribute Delete
1.49 kB
---
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`](https://huggingface.co/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
```python
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*