cat-searcher
commited on
Commit
•
429bec6
1
Parent(s):
4c62b8e
End of training
Browse files- README.md +7 -1
- config.json +1 -1
README.md
CHANGED
@@ -2,9 +2,15 @@
|
|
2 |
license: gemma
|
3 |
base_model: google/gemma-1.1-2b-it
|
4 |
tags:
|
|
|
5 |
- trl
|
6 |
- dpo
|
7 |
- generated_from_trainer
|
|
|
|
|
|
|
|
|
|
|
8 |
model-index:
|
9 |
- name: gemma-1.1-2b-it-sppo-iter0-evol-mixed
|
10 |
results: []
|
@@ -16,7 +22,7 @@ should probably proofread and complete it, then remove this comment. -->
|
|
16 |
[<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="200" height="32"/>](https://wandb.ai/the-dream-machine/huggingface/runs/ciqulebv)
|
17 |
# gemma-1.1-2b-it-sppo-iter0-evol-mixed
|
18 |
|
19 |
-
This model is a fine-tuned version of [google/gemma-1.1-2b-it](https://huggingface.co/google/gemma-1.1-2b-it) on
|
20 |
|
21 |
## Model description
|
22 |
|
|
|
2 |
license: gemma
|
3 |
base_model: google/gemma-1.1-2b-it
|
4 |
tags:
|
5 |
+
- alignment-handbook
|
6 |
- trl
|
7 |
- dpo
|
8 |
- generated_from_trainer
|
9 |
+
- trl
|
10 |
+
- dpo
|
11 |
+
- generated_from_trainer
|
12 |
+
datasets:
|
13 |
+
- cat-searcher/responses-gemma-1.1-2b-it-split-0-evol-mixed-pair
|
14 |
model-index:
|
15 |
- name: gemma-1.1-2b-it-sppo-iter0-evol-mixed
|
16 |
results: []
|
|
|
22 |
[<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="200" height="32"/>](https://wandb.ai/the-dream-machine/huggingface/runs/ciqulebv)
|
23 |
# gemma-1.1-2b-it-sppo-iter0-evol-mixed
|
24 |
|
25 |
+
This model is a fine-tuned version of [google/gemma-1.1-2b-it](https://huggingface.co/google/gemma-1.1-2b-it) on the cat-searcher/responses-gemma-1.1-2b-it-split-0-evol-mixed-pair dataset.
|
26 |
|
27 |
## Model description
|
28 |
|
config.json
CHANGED
@@ -23,6 +23,6 @@
|
|
23 |
"rope_theta": 10000.0,
|
24 |
"torch_dtype": "bfloat16",
|
25 |
"transformers_version": "4.42.4",
|
26 |
-
"use_cache":
|
27 |
"vocab_size": 256000
|
28 |
}
|
|
|
23 |
"rope_theta": 10000.0,
|
24 |
"torch_dtype": "bfloat16",
|
25 |
"transformers_version": "4.42.4",
|
26 |
+
"use_cache": true,
|
27 |
"vocab_size": 256000
|
28 |
}
|