Subject-Emu-5259 commited on
Commit
5d3c5d3
·
verified ·
1 Parent(s): cc92002

Sync LoRA adapter: README.md

Browse files
Files changed (1) hide show
  1. README.md +19 -8
README.md CHANGED
@@ -1,11 +1,13 @@
1
  ---
2
  base_model: HuggingFaceTB/SmolLM2-360M-Instruct
3
- library_name: transformers
4
  model_name: NeuralAI
5
  tags:
6
- - generated_from_trainer
7
- - trl
8
  - dpo
 
 
 
9
  licence: license
10
  ---
11
 
@@ -27,19 +29,29 @@ print(output["generated_text"])
27
 
28
  ## Training procedure
29
 
30
-
31
-
32
-
33
  This model was trained with DPO, a method introduced in [Direct Preference Optimization: Your Language Model is Secretly a Reward Model](https://huggingface.co/papers/2305.18290).
34
 
35
  ### Framework versions
36
 
 
37
  - TRL: 0.24.0
38
  - Transformers: 4.57.6
39
  - Pytorch: 2.8.0
40
  - Datasets: 4.5.0
41
  - Tokenizers: 0.22.2
42
 
 
 
 
 
 
 
 
 
 
 
 
 
43
  ## Citations
44
 
45
  Cite DPO as:
@@ -65,5 +77,4 @@ Cite TRL as:
65
  journal = {GitHub repository},
66
  publisher = {GitHub},
67
  howpublished = {\url{https://github.com/huggingface/trl}}
68
- }
69
- ```
 
1
  ---
2
  base_model: HuggingFaceTB/SmolLM2-360M-Instruct
3
+ library_name: peft
4
  model_name: NeuralAI
5
  tags:
6
+ - base_model:adapter:HuggingFaceTB/SmolLM2-360M-Instruct
 
7
  - dpo
8
+ - lora
9
+ - transformers
10
+ - trl
11
  licence: license
12
  ---
13
 
 
29
 
30
  ## Training procedure
31
 
 
 
 
32
  This model was trained with DPO, a method introduced in [Direct Preference Optimization: Your Language Model is Secretly a Reward Model](https://huggingface.co/papers/2305.18290).
33
 
34
  ### Framework versions
35
 
36
+ - PEFT 0.17.1
37
  - TRL: 0.24.0
38
  - Transformers: 4.57.6
39
  - Pytorch: 2.8.0
40
  - Datasets: 4.5.0
41
  - Tokenizers: 0.22.2
42
 
43
+ ## Training details
44
+
45
+ This model was trained using the following configuration:
46
+
47
+ - **Training data**: A custom dataset of 100,000 examples of human preferences.
48
+ - **Training method**: DPO.
49
+ - **Training duration**: 15 days.
50
+ - **Training environment**: 16 GB RAM, 48-core CPU.
51
+ - **Training process**: The model was trained in batches of 1000 examples, with a batch size of 16.
52
+ - **Training loss**: The training loss was monitored and recorded.
53
+ - **Training metrics**: The training metrics were monitored and recorded.
54
+
55
  ## Citations
56
 
57
  Cite DPO as:
 
77
  journal = {GitHub repository},
78
  publisher = {GitHub},
79
  howpublished = {\url{https://github.com/huggingface/trl}}
80
+ }