poisson-fish
commited on
Commit
·
3cb457a
1
Parent(s):
05f1e5d
Upload folder using huggingface_hub
Browse files- README.md +93 -0
- adapter_config.json +23 -0
- adapter_model.bin +3 -0
- config.json +28 -0
- merges.txt +0 -0
- special_tokens_map.json +34 -0
- tokenizer.json +0 -0
- tokenizer_config.json +58 -0
- vocab.json +0 -0
README.md
ADDED
@@ -0,0 +1,93 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
license: wtfpl
|
3 |
+
base_model: CausalLM/7B
|
4 |
+
tags:
|
5 |
+
- generated_from_trainer
|
6 |
+
model-index:
|
7 |
+
- name: workspace/causal-dolphin-v0.1
|
8 |
+
results: []
|
9 |
+
---
|
10 |
+
|
11 |
+
[<img src="https://raw.githubusercontent.com/OpenAccess-AI-Collective/axolotl/main/image/axolotl-badge-web.png" alt="Built with Axolotl" width="200" height="32"/>](https://github.com/OpenAccess-AI-Collective/axolotl)
|
12 |
+
# Causal-Dolphin-Agent-v0.1
|
13 |
+
|
14 |
+
This model is a LoRA fine-tune of [CausalLM/7B](https://huggingface.co/CausalLM/7B) on Eric's wonderful [Dolphin](https://huggingface.co/datasets/ehartford/dolphin) dataset, with [THUDM/AgentInstruct](https://huggingface.co/datasets/THUDM/AgentInstruct) mixed in both training runs.
|
15 |
+
|
16 |
+
Causal-Dolphin-Agent was trained for 3 epochs on 5 million GPT3.5 augmented FLAN instructions & AgentInstruct dataset in ChatML format. It was then trained a further 3 epochs on 1 million GPT4 augmented FLAN instructions with AgentInstruct shuffled in as well.
|
17 |
+
|
18 |
+
It achieves the following results on the evaluation set:
|
19 |
+
- Loss: 2.8435
|
20 |
+
|
21 |
+
## Prompt Format
|
22 |
+
|
23 |
+
Causal-Dolphin-Agent uses ChatML as the prompt format:
|
24 |
+
```<|im_start|>system
|
25 |
+
You are Dolphin, a helpful AI assistant.<|im_end|>
|
26 |
+
<|im_start|>user
|
27 |
+
If Danny owns a bike, then Edward owns a bike. If Edward owns a bike, then Freddy owns a bike. If Danny owns a bike, which of the following statements must be true? Let's think step by step.
|
28 |
+
|
29 |
+
I. Edward owns a bike.
|
30 |
+
II. Freddy owns a bike.
|
31 |
+
III. Freddy does not own a bike.
|
32 |
+
|
33 |
+
Choose one answer:
|
34 |
+
I only
|
35 |
+
II only
|
36 |
+
III only
|
37 |
+
I and II only
|
38 |
+
I and III only
|
39 |
+
<|im_end|>
|
40 |
+
<|im_start|>assistant```
|
41 |
+
|
42 |
+
## Training and evaluation data
|
43 |
+
|
44 |
+
[ehartford/dolphin](https://huggingface.co/datasets/ehartford/dolphin)
|
45 |
+
[THUDM/AgentInstruct](https://huggingface.co/datasets/THUDM/AgentInstruct)
|
46 |
+
|
47 |
+
## Training procedure
|
48 |
+
Causal-Dolphin-Agent was trained for 3 epochs on 5 million GPT3.5 augmented FLAN instructions & AgentInstruct dataset in ChatML format. It was then trained a further 3 epochs on 1 million GPT4 augmented FLAN instructions with AgentInstruct shuffled in as well.
|
49 |
+
|
50 |
+
### Training hyperparameters
|
51 |
+
|
52 |
+
The following hyperparameters were used during training:
|
53 |
+
- learning_rate: 6e-06
|
54 |
+
- train_batch_size: 1
|
55 |
+
- eval_batch_size: 1
|
56 |
+
- seed: 42
|
57 |
+
- optimizer: Adam with betas=(0.9,0.95) and epsilon=1e-05
|
58 |
+
- lr_scheduler_type: cosine
|
59 |
+
- lr_scheduler_warmup_steps: 100
|
60 |
+
- num_epochs: 3
|
61 |
+
|
62 |
+
### Training results
|
63 |
+
|
64 |
+
| Training Loss | Epoch | Step | Validation Loss |
|
65 |
+
|:-------------:|:-----:|:----:|:---------------:|
|
66 |
+
| 2.7774 | 0.0 | 1 | 5.1009 |
|
67 |
+
| 3.2798 | 0.15 | 46 | 5.1010 |
|
68 |
+
| 2.0722 | 0.3 | 92 | 5.0489 |
|
69 |
+
| 2.5919 | 0.45 | 138 | 4.8834 |
|
70 |
+
| 2.0011 | 0.6 | 184 | 4.6678 |
|
71 |
+
| 1.3733 | 0.75 | 230 | 4.4628 |
|
72 |
+
| 1.7321 | 0.9 | 276 | 4.2757 |
|
73 |
+
| 1.3994 | 1.05 | 322 | 4.1029 |
|
74 |
+
| 1.2308 | 1.2 | 368 | 3.8916 |
|
75 |
+
| 0.8229 | 1.35 | 414 | 3.6451 |
|
76 |
+
| 0.9592 | 1.5 | 460 | 3.4106 |
|
77 |
+
| 0.8528 | 1.65 | 506 | 3.2250 |
|
78 |
+
| 0.7362 | 1.8 | 552 | 3.0852 |
|
79 |
+
| 0.8077 | 1.95 | 598 | 2.9881 |
|
80 |
+
| 0.6912 | 2.1 | 644 | 2.9315 |
|
81 |
+
| 0.7776 | 2.25 | 690 | 2.8911 |
|
82 |
+
| 0.6916 | 2.41 | 736 | 2.8678 |
|
83 |
+
| 0.8674 | 2.56 | 782 | 2.8534 |
|
84 |
+
| 0.7797 | 2.71 | 828 | 2.8545 |
|
85 |
+
| 0.6838 | 2.86 | 874 | 2.8435 |
|
86 |
+
|
87 |
+
|
88 |
+
### Framework versions
|
89 |
+
|
90 |
+
- Transformers 4.34.1
|
91 |
+
- Pytorch 2.1.0+cu121
|
92 |
+
- Datasets 2.14.6
|
93 |
+
- Tokenizers 0.14.1
|
adapter_config.json
ADDED
@@ -0,0 +1,23 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"alpha_pattern": {},
|
3 |
+
"auto_mapping": null,
|
4 |
+
"base_model_name_or_path": "CausalLM/7B",
|
5 |
+
"bias": "none",
|
6 |
+
"fan_in_fan_out": null,
|
7 |
+
"inference_mode": true,
|
8 |
+
"init_lora_weights": true,
|
9 |
+
"layers_pattern": null,
|
10 |
+
"layers_to_transform": null,
|
11 |
+
"lora_alpha": 16,
|
12 |
+
"lora_dropout": 0.05,
|
13 |
+
"modules_to_save": null,
|
14 |
+
"peft_type": "LORA",
|
15 |
+
"r": 32,
|
16 |
+
"rank_pattern": {},
|
17 |
+
"revision": null,
|
18 |
+
"target_modules": [
|
19 |
+
"v_proj",
|
20 |
+
"q_proj"
|
21 |
+
],
|
22 |
+
"task_type": "CAUSAL_LM"
|
23 |
+
}
|
adapter_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:22980146de9b7323bc4ca16384b2dc25e239d966273a317a7d06c564fdacbae9
|
3 |
+
size 67155338
|
config.json
ADDED
@@ -0,0 +1,28 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_name_or_path": "CausalLM/7B",
|
3 |
+
"architectures": [
|
4 |
+
"LlamaForCausalLM"
|
5 |
+
],
|
6 |
+
"attention_bias": false,
|
7 |
+
"bos_token_id": 151643,
|
8 |
+
"eos_token_id": 151643,
|
9 |
+
"hidden_act": "silu",
|
10 |
+
"hidden_size": 4096,
|
11 |
+
"initializer_range": 0.02,
|
12 |
+
"intermediate_size": 11008,
|
13 |
+
"max_position_embeddings": 8192,
|
14 |
+
"model_type": "llama",
|
15 |
+
"num_attention_heads": 32,
|
16 |
+
"num_hidden_layers": 32,
|
17 |
+
"num_key_value_heads": 32,
|
18 |
+
"pad_token_id": 151643,
|
19 |
+
"pretraining_tp": 1,
|
20 |
+
"rms_norm_eps": 1e-05,
|
21 |
+
"rope_scaling": null,
|
22 |
+
"rope_theta": 10000.0,
|
23 |
+
"tie_word_embeddings": false,
|
24 |
+
"torch_dtype": "bfloat16",
|
25 |
+
"transformers_version": "4.34.1",
|
26 |
+
"use_cache": false,
|
27 |
+
"vocab_size": 151936
|
28 |
+
}
|
merges.txt
ADDED
The diff for this file is too large to render.
See raw diff
|
|
special_tokens_map.json
ADDED
@@ -0,0 +1,34 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"additional_special_tokens": [
|
3 |
+
"<|im_start|>",
|
4 |
+
"<|im_end|>"
|
5 |
+
],
|
6 |
+
"bos_token": {
|
7 |
+
"content": "<s>",
|
8 |
+
"lstrip": false,
|
9 |
+
"normalized": false,
|
10 |
+
"rstrip": false,
|
11 |
+
"single_word": false
|
12 |
+
},
|
13 |
+
"eos_token": {
|
14 |
+
"content": "<|im_end|>",
|
15 |
+
"lstrip": false,
|
16 |
+
"normalized": false,
|
17 |
+
"rstrip": false,
|
18 |
+
"single_word": false
|
19 |
+
},
|
20 |
+
"pad_token": {
|
21 |
+
"content": "<|endoftext|>",
|
22 |
+
"lstrip": false,
|
23 |
+
"normalized": false,
|
24 |
+
"rstrip": false,
|
25 |
+
"single_word": false
|
26 |
+
},
|
27 |
+
"unk_token": {
|
28 |
+
"content": "<unk>",
|
29 |
+
"lstrip": false,
|
30 |
+
"normalized": false,
|
31 |
+
"rstrip": false,
|
32 |
+
"single_word": false
|
33 |
+
}
|
34 |
+
}
|
tokenizer.json
ADDED
The diff for this file is too large to render.
See raw diff
|
|
tokenizer_config.json
ADDED
@@ -0,0 +1,58 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"add_bos_token": false,
|
3 |
+
"add_prefix_space": false,
|
4 |
+
"added_tokens_decoder": {
|
5 |
+
"128244": {
|
6 |
+
"content": "<unk>",
|
7 |
+
"lstrip": false,
|
8 |
+
"normalized": false,
|
9 |
+
"rstrip": false,
|
10 |
+
"single_word": false,
|
11 |
+
"special": true
|
12 |
+
},
|
13 |
+
"128245": {
|
14 |
+
"content": "<s>",
|
15 |
+
"lstrip": false,
|
16 |
+
"normalized": false,
|
17 |
+
"rstrip": false,
|
18 |
+
"single_word": false,
|
19 |
+
"special": true
|
20 |
+
},
|
21 |
+
"151643": {
|
22 |
+
"content": "<|endoftext|>",
|
23 |
+
"lstrip": false,
|
24 |
+
"normalized": false,
|
25 |
+
"rstrip": false,
|
26 |
+
"single_word": false,
|
27 |
+
"special": true
|
28 |
+
},
|
29 |
+
"151644": {
|
30 |
+
"content": "<|im_start|>",
|
31 |
+
"lstrip": false,
|
32 |
+
"normalized": false,
|
33 |
+
"rstrip": false,
|
34 |
+
"single_word": false,
|
35 |
+
"special": false
|
36 |
+
},
|
37 |
+
"151645": {
|
38 |
+
"content": "<|im_end|>",
|
39 |
+
"lstrip": false,
|
40 |
+
"normalized": false,
|
41 |
+
"rstrip": false,
|
42 |
+
"single_word": false,
|
43 |
+
"special": false
|
44 |
+
}
|
45 |
+
},
|
46 |
+
"additional_special_tokens": [
|
47 |
+
"<|im_start|>",
|
48 |
+
"<|im_end|>"
|
49 |
+
],
|
50 |
+
"bos_token": "<s>",
|
51 |
+
"clean_up_tokenization_spaces": true,
|
52 |
+
"eos_token": "<|im_end|>",
|
53 |
+
"errors": "replace",
|
54 |
+
"model_max_length": 1000000000000000019884624838656,
|
55 |
+
"pad_token": "<|endoftext|>",
|
56 |
+
"tokenizer_class": "GPT2Tokenizer",
|
57 |
+
"unk_token": "<unk>"
|
58 |
+
}
|
vocab.json
ADDED
The diff for this file is too large to render.
See raw diff
|
|