cat-searcher commited on
Commit
429bec6
1 Parent(s): 4c62b8e

End of training

Browse files
Files changed (2) hide show
  1. README.md +7 -1
  2. 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 an unknown dataset.
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": false,
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
  }