theprint commited on
Commit
2ba7e02
·
verified ·
1 Parent(s): 4538082

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +63 -0
README.md ADDED
@@ -0,0 +1,63 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: unsloth/Qwen3.5-2B
3
+ datasets:
4
+ - OpceanAI/sota-coding
5
+ tags:
6
+ - fine-tuned
7
+ - lora
8
+ - sft
9
+ - auto-sft
10
+ language:
11
+ - en
12
+ library_name: transformers
13
+ ---
14
+
15
+ # SoCode-v1-2B
16
+
17
+ 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.
18
+
19
+ 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.
20
+
21
+ ## Model Details
22
+
23
+ | Property | Value |
24
+ |---|---|
25
+ | Base model | `unsloth/Qwen3.5-2B` |
26
+ | Training data | `OpceanAI/sota-coding` |
27
+ | Fine-tuning epochs | 1 |
28
+ | Fine-tuning date | 2026-07-21 |
29
+ | Fine-tuning method | LoRA (merged to full 16-bit) |
30
+
31
+ ## Training Hyperparameters
32
+
33
+ ### LoRA
34
+
35
+ | Parameter | Value |
36
+ |---|---|
37
+ | `r` | `64` |
38
+ | `alpha` | `256` |
39
+ | `dropout` | `0.07` |
40
+ | `target_modules` | `['q_proj', 'v_proj']` |
41
+
42
+ ### Training
43
+
44
+ | Parameter | Value |
45
+ |---|---|
46
+ | `learning_rate` | `0.0002` |
47
+ | `batch_size` | `1` |
48
+ | `gradient_accumulation_steps` | `8` |
49
+ | `warmup_ratio` | `0.1` |
50
+ | `max_seq_length` | `512` |
51
+ | `quantization` | `none` |
52
+
53
+ ## Usage
54
+
55
+ ```python
56
+ from transformers import AutoModelForCausalLM, AutoTokenizer
57
+
58
+ model = AutoModelForCausalLM.from_pretrained("theprint/SoCode-v1-2B")
59
+ tokenizer = AutoTokenizer.from_pretrained("theprint/SoCode-v1-2B")
60
+ ```
61
+
62
+ ---
63
+ *Generated by Auto-SFT*