theprint commited on
Commit
395bea7
·
verified ·
1 Parent(s): 5cfa76c

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +61 -0
README.md ADDED
@@ -0,0 +1,61 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: unsloth/Qwen3.5-2B
3
+ tags:
4
+ - fine-tuned
5
+ - lora
6
+ - sft
7
+ - auto-sft
8
+ language:
9
+ - en
10
+ library_name: transformers
11
+ ---
12
+
13
+ # ProgramManager-v1-2B
14
+
15
+ A fine-tuned version of [`unsloth/Qwen3.5-2B`](https://huggingface.co/unsloth/Qwen3.5-2B) trained on **TechPM Combined 6.40k sharegpt** data using [Auto-SFT](https://github.com/theprint/auto-sft) — an automated hyperparameter search and supervised fine-tuning pipeline.
16
+
17
+ The base model was adapted to follow the style and content of the `TechPM Combined 6.40k sharegpt` dataset. Expect improved performance on tasks similar to those represented in the training data.
18
+
19
+ ## Model Details
20
+
21
+ | Property | Value |
22
+ |---|---|
23
+ | Base model | `unsloth/Qwen3.5-2B` |
24
+ | Training data | `data/TechPM-Combined-6.40k-sharegpt.json` |
25
+ | Fine-tuning epochs | 2 |
26
+ | Fine-tuning date | 2026-08-26 |
27
+ | Fine-tuning method | LoRA (merged to full 16-bit) |
28
+
29
+ ## Training Hyperparameters
30
+
31
+ ### LoRA
32
+
33
+ | Parameter | Value |
34
+ |---|---|
35
+ | `r` | `4` |
36
+ | `alpha` | `4` |
37
+ | `dropout` | `0.07` |
38
+ | `target_modules` | `['q_proj', 'v_proj', 'k_proj', 'o_proj']` |
39
+
40
+ ### Training
41
+
42
+ | Parameter | Value |
43
+ |---|---|
44
+ | `learning_rate` | `1e-05` |
45
+ | `batch_size` | `1` |
46
+ | `gradient_accumulation_steps` | `8` |
47
+ | `warmup_ratio` | `0.05` |
48
+ | `max_seq_length` | `512` |
49
+ | `quantization` | `none` |
50
+
51
+ ## Usage
52
+
53
+ ```python
54
+ from transformers import AutoModelForCausalLM, AutoTokenizer
55
+
56
+ model = AutoModelForCausalLM.from_pretrained("theprint/ProgramManager-v1-2B")
57
+ tokenizer = AutoTokenizer.from_pretrained("theprint/ProgramManager-v1-2B")
58
+ ```
59
+
60
+ ---
61
+ *Generated by [Auto-SFT](https://github.com/theprint/auto-sft) — automated LoRA fine-tuning with hyperparameter search.*