lole25 commited on
Commit
67c2b0e
1 Parent(s): a628f21

End of training

Browse files
Files changed (2) hide show
  1. README.md +4 -2
  2. config.json +1 -1
README.md CHANGED
@@ -2,11 +2,13 @@
2
  license: mit
3
  base_model: microsoft/phi-2
4
  tags:
 
 
5
  - trl
6
  - sft
7
  - generated_from_trainer
8
  datasets:
9
- - generator
10
  model-index:
11
  - name: phi-2-sft-ultrachat-full
12
  results: []
@@ -17,7 +19,7 @@ should probably proofread and complete it, then remove this comment. -->
17
 
18
  # phi-2-sft-ultrachat-full
19
 
20
- This model is a fine-tuned version of [microsoft/phi-2](https://huggingface.co/microsoft/phi-2) on the generator dataset.
21
  It achieves the following results on the evaluation set:
22
  - Loss: 1.1928
23
 
 
2
  license: mit
3
  base_model: microsoft/phi-2
4
  tags:
5
+ - alignment-handbook
6
+ - generated_from_trainer
7
  - trl
8
  - sft
9
  - generated_from_trainer
10
  datasets:
11
+ - HuggingFaceH4/ultrachat_200k
12
  model-index:
13
  - name: phi-2-sft-ultrachat-full
14
  results: []
 
19
 
20
  # phi-2-sft-ultrachat-full
21
 
22
+ This model is a fine-tuned version of [microsoft/phi-2](https://huggingface.co/microsoft/phi-2) on the HuggingFaceH4/ultrachat_200k dataset.
23
  It achieves the following results on the evaluation set:
24
  - Loss: 1.1928
25
 
config.json CHANGED
@@ -29,6 +29,6 @@
29
  "tie_word_embeddings": false,
30
  "torch_dtype": "bfloat16",
31
  "transformers_version": "4.36.2",
32
- "use_cache": false,
33
  "vocab_size": 51200
34
  }
 
29
  "tie_word_embeddings": false,
30
  "torch_dtype": "bfloat16",
31
  "transformers_version": "4.36.2",
32
+ "use_cache": true,
33
  "vocab_size": 51200
34
  }