File size: 1,444 Bytes
849ba45
859f70d
849ba45
 
859f70d
849ba45
859f70d
849ba45
859f70d
849ba45
859f70d
 
1ac3980
859f70d
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
---
license: "apache-2.0"
---

*This model was trained as part of a series of experiments testing the performance of pure DPO vs SFT vs ORPO, all supported by Unsloth/Huggingface TRL.*

Note: Extremely buggy, not recommended for use. However, it didn't massively overfit like #3, so it could be usable still. 

The training was somewhat unstable, so the optimal bound for LR seems to be around [1e-5, 1e-4].

**Benchmarks**

For some reason the OpenLLM leaderboard refuses to bench this model, so I guess we will never know how well it performs.

**Training Details**

Duration: ~10-12 hours on one Kaggle T4 with Unsloth

Model: https://huggingface.co/unsloth/mistral-7b-v0.2-bnb-4bit

Dataset: https://huggingface.co/datasets/argilla/dpo-mix-7k 

Rank: 8 

Alpha: 16 

Learning rate: 1e-4

Beta: 0.1 

Batch size: 8 

Epochs: 1

Learning rate scheduler: Linear

Prompt Format: ChatML
```
<|im_start|>system
You are a helpful assistant.<|im_end|>
<|im_start|>user
Why is the sky blue?<|im_end|>
<|im_start|>assistant
```


**WanDB Reports**

![image/png](https://cdn-uploads.huggingface.co/production/uploads/65a5c0e82823ba72ed2cee7d/Ww-urn-b22jj2sr735rNs.png)

![image/png](https://cdn-uploads.huggingface.co/production/uploads/65a5c0e82823ba72ed2cee7d/hVzS-W9SGA8TZn65ixF84.png)

[<img src="https://raw.githubusercontent.com/unslothai/unsloth/main/images/unsloth%20made%20with%20love.png" width="200"/>](https://github.com/unslothai/unsloth)