Manually update README.md
Browse files
README.md
CHANGED
@@ -15,26 +15,48 @@ model-index:
|
|
15 |
results: []
|
16 |
---
|
17 |
|
18 |
-
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
|
19 |
-
should probably proofread and complete it, then remove this comment. -->
|
20 |
-
|
21 |
# zephyr-7b-sft-qlora
|
22 |
|
23 |
-
This model is a fine-tuned version of [mistralai/Mistral-7B-v0.1](https://huggingface.co/mistralai/
|
|
|
24 |
It achieves the following results on the evaluation set:
|
25 |
- Loss: 0.9523
|
26 |
|
27 |
## Model description
|
28 |
|
29 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
30 |
|
31 |
## Intended uses & limitations
|
32 |
|
33 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
34 |
|
35 |
## Training and evaluation data
|
36 |
|
37 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
38 |
|
39 |
## Training procedure
|
40 |
|
|
|
15 |
results: []
|
16 |
---
|
17 |
|
|
|
|
|
|
|
18 |
# zephyr-7b-sft-qlora
|
19 |
|
20 |
+
This model is a fine-tuned version of [mistralai/Mistral-7B-v0.1](https://huggingface.co/mistralai/Mist\
|
21 |
+
ral-7B-v0.1) on the HuggingFaceH4/ultrachat_200k dataset.
|
22 |
It achieves the following results on the evaluation set:
|
23 |
- Loss: 0.9523
|
24 |
|
25 |
## Model description
|
26 |
|
27 |
+
QLoRA SFT via
|
28 |
+
```
|
29 |
+
# Step 1 - SFT
|
30 |
+
ACCELERATE_LOG_LEVEL=info accelerate launch --config_file recipes/accelerate_configs/multi_gpu.yaml --n\
|
31 |
+
um_processes=1 scripts/run_sft.py recipes/zephyr-7b-beta/sft/config_qlora.yaml --load_in_4bit=true
|
32 |
+
```
|
33 |
+
see https://github.com/huggingface/alignment-handbook/blob/main/recipes/zephyr-7b-beta/README.md
|
34 |
|
35 |
## Intended uses & limitations
|
36 |
|
37 |
+
```
|
38 |
+
chat_template: "{% for message in messages %}\n{% if message['role'] == 'user' %}\n{{ '<|user|>\n' + me\
|
39 |
+
ssage['content'] + eos_token }}\n{% elif message['role'] == 'system' %}\n{{ '<|system|>\n' + message['c\
|
40 |
+
ontent'] + eos_token }}\n{% elif message['role'] == 'assistant' %}\n{{ '<|assistant|>\n' + message['co\
|
41 |
+
ntent'] + eos_token }}\n{% endif %}\n{% if loop.last and add_generation_prompt %}\n{{ '<|assistant|>' }\
|
42 |
+
}\n{% endif %}\n{% endfor %}"
|
43 |
+
```
|
44 |
+
|
45 |
+
see https://github.com/huggingface/alignment-handbook/blob/main/recipes/zephyr-7b-beta/sft/config_qlora\
|
46 |
+
.yaml
|
47 |
|
48 |
## Training and evaluation data
|
49 |
|
50 |
+
```
|
51 |
+
dataset_mixer:
|
52 |
+
HuggingFaceH4/ultrachat_200k: 1.0
|
53 |
+
dataset_splits:
|
54 |
+
- train_sft
|
55 |
+
- test_sft
|
56 |
+
```
|
57 |
+
|
58 |
+
see https://github.com/huggingface/alignment-handbook/blob/main/recipes/zephyr-7b-beta/sft/config_qlora\
|
59 |
+
.yaml
|
60 |
|
61 |
## Training procedure
|
62 |
|