mrcuddle commited on
Commit
ec5c560
·
verified ·
1 Parent(s): 0253c57

Training in progress, epoch 0

Browse files
README.md ADDED
@@ -0,0 +1,117 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: other
4
+ base_model: mrcuddle/tiny-darkllama
5
+ tags:
6
+ - generated_from_trainer
7
+ datasets:
8
+ - Nitral-AI/Reddit-NSFW-Writing_Prompts_ShareGPT
9
+ model-index:
10
+ - name: model-out
11
+ results: []
12
+ ---
13
+
14
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
15
+ should probably proofread and complete it, then remove this comment. -->
16
+
17
+ [<img src="https://raw.githubusercontent.com/axolotl-ai-cloud/axolotl/main/image/axolotl-badge-web.png" alt="Built with Axolotl" width="200" height="32"/>](https://github.com/axolotl-ai-cloud/axolotl)
18
+ <details><summary>See axolotl config</summary>
19
+
20
+ axolotl version: `0.6.0`
21
+ ```yaml
22
+ base_model: mrcuddle/tiny-darkllama
23
+ bf16: auto
24
+ datasets:
25
+ - chat_template: tokenizer_default
26
+ field_messages: conversations
27
+ message_field_content: value
28
+ message_field_role: from
29
+ path: Nitral-AI/Reddit-NSFW-Writing_Prompts_ShareGPT
30
+ split: train
31
+ type: chat_template
32
+ debug: null
33
+ deepspeed: null
34
+ early_stopping_patience: null
35
+ eval_sample_packing: true
36
+ eval_table_size: null
37
+ evals_per_epoch: 1
38
+ flash_attention: false
39
+ fp16: null
40
+ fsdp: null
41
+ fsdp_config: null
42
+ gradient_accumulation_steps: 2
43
+ gradient_checkpointing: true
44
+ group_by_length: false
45
+ learning_rate: 2e-5
46
+ load_in_4bit: false
47
+ load_in_8bit: false
48
+ logging_steps: 1
49
+ lr_scheduler: cosine
50
+ max_steps: 25
51
+ micro_batch_size: 1
52
+ num_epochs: 1
53
+ optimizer: paged_adamw_8bit
54
+ pad_to_sequence_len: true
55
+ resume_from_checkpoint: null
56
+ sample_packing: true
57
+ saves_per_epoch: 1
58
+ sdp_attention: true
59
+ sequence_len: 2048
60
+ special_tokens:
61
+ pad_token: <|end_of_text|>
62
+ strict: false
63
+ tf32: false
64
+ train_on_inputs: false
65
+ warmup_steps: 1
66
+ weight_decay: 0.0
67
+ xformers_attention: null
68
+
69
+ ```
70
+
71
+ </details><br>
72
+
73
+ # model-out
74
+
75
+ This model is a fine-tuned version of [mrcuddle/tiny-darkllama](https://huggingface.co/mrcuddle/tiny-darkllama) on the Nitral-AI/Reddit-NSFW-Writing_Prompts_ShareGPT dataset.
76
+
77
+ ## Model description
78
+
79
+ More information needed
80
+
81
+ ## Intended uses & limitations
82
+
83
+ More information needed
84
+
85
+ ## Training and evaluation data
86
+
87
+ More information needed
88
+
89
+ ## Training procedure
90
+
91
+ ### Training hyperparameters
92
+
93
+ The following hyperparameters were used during training:
94
+ - learning_rate: 2e-05
95
+ - train_batch_size: 1
96
+ - eval_batch_size: 1
97
+ - seed: 42
98
+ - distributed_type: multi-GPU
99
+ - num_devices: 2
100
+ - gradient_accumulation_steps: 2
101
+ - total_train_batch_size: 4
102
+ - total_eval_batch_size: 2
103
+ - optimizer: Use OptimizerNames.PAGED_ADAMW_8BIT with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
104
+ - lr_scheduler_type: cosine
105
+ - lr_scheduler_warmup_steps: 2
106
+ - training_steps: 25
107
+
108
+ ### Training results
109
+
110
+
111
+
112
+ ### Framework versions
113
+
114
+ - Transformers 4.47.1
115
+ - Pytorch 2.5.1+cu124
116
+ - Datasets 3.1.0
117
+ - Tokenizers 0.21.0
adapter_config.json ADDED
@@ -0,0 +1,40 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alpha_pattern": {},
3
+ "auto_mapping": null,
4
+ "base_model_name_or_path": "mrcuddle/tiny-darkllama",
5
+ "bias": "none",
6
+ "eva_config": null,
7
+ "exclude_modules": null,
8
+ "fan_in_fan_out": null,
9
+ "inference_mode": true,
10
+ "init_lora_weights": true,
11
+ "layer_replication": null,
12
+ "layers_pattern": null,
13
+ "layers_to_transform": null,
14
+ "loftq_config": {},
15
+ "lora_alpha": 16,
16
+ "lora_bias": false,
17
+ "lora_dropout": 0.05,
18
+ "megatron_config": null,
19
+ "megatron_core": "megatron.core",
20
+ "modules_to_save": [
21
+ "embed_tokens",
22
+ "lm_head"
23
+ ],
24
+ "peft_type": "LORA",
25
+ "r": 32,
26
+ "rank_pattern": {},
27
+ "revision": null,
28
+ "target_modules": [
29
+ "v_proj",
30
+ "o_proj",
31
+ "gate_proj",
32
+ "q_proj",
33
+ "down_proj",
34
+ "up_proj",
35
+ "k_proj"
36
+ ],
37
+ "task_type": "CAUSAL_LM",
38
+ "use_dora": false,
39
+ "use_rslora": false
40
+ }
adapter_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:70d27281026c9ebcf91bc9742862eac3c83bc4e15483456cbb44c45debda3007
3
+ size 625342634
adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1873521df19237be0c6fa0138e1fbec82fe1754e6728cd69ee4e81c89be8bc2b
3
+ size 887423664
added_tokens.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ {
2
+ "<|end_of_text|>": 32000
3
+ }
config.json ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "mrcuddle/tiny-darkllama",
3
+ "architectures": [
4
+ "LlamaForCausalLM"
5
+ ],
6
+ "attention_bias": false,
7
+ "attention_dropout": 0.0,
8
+ "bos_token_id": 1,
9
+ "eos_token_id": 2,
10
+ "head_dim": 64,
11
+ "hidden_act": "silu",
12
+ "hidden_size": 2048,
13
+ "initializer_range": 0.02,
14
+ "intermediate_size": 5632,
15
+ "max_position_embeddings": 2048,
16
+ "mlp_bias": false,
17
+ "model_type": "llama",
18
+ "num_attention_heads": 32,
19
+ "num_hidden_layers": 22,
20
+ "num_key_value_heads": 4,
21
+ "pretraining_tp": 1,
22
+ "rms_norm_eps": 1e-05,
23
+ "rope_scaling": null,
24
+ "rope_theta": 10000.0,
25
+ "tie_word_embeddings": false,
26
+ "torch_dtype": "float32",
27
+ "transformers_version": "4.47.1",
28
+ "use_cache": false,
29
+ "vocab_size": 32001
30
+ }
generation_config.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token_id": 1,
3
+ "do_sample": true,
4
+ "eos_token_id": 2,
5
+ "max_length": 2048,
6
+ "pad_token_id": 0,
7
+ "transformers_version": "4.47.1"
8
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d47d298beba548abe49d6e72c37f6461157fb40f1da332d8a40261126df4b376
3
+ size 2462464504
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a4550a2e5d4d50ac8865efcf01a3965049a310bc1c53b858b9b72dcda661654c
3
+ size 2462509422
special_tokens_map.json ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": {
3
+ "content": "<s>",
4
+ "lstrip": false,
5
+ "normalized": false,
6
+ "rstrip": false,
7
+ "single_word": false
8
+ },
9
+ "eos_token": {
10
+ "content": "</s>",
11
+ "lstrip": false,
12
+ "normalized": false,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "pad_token": {
17
+ "content": "<|end_of_text|>",
18
+ "lstrip": false,
19
+ "normalized": false,
20
+ "rstrip": false,
21
+ "single_word": false
22
+ },
23
+ "unk_token": {
24
+ "content": "<unk>",
25
+ "lstrip": false,
26
+ "normalized": false,
27
+ "rstrip": false,
28
+ "single_word": false
29
+ }
30
+ }
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer.model ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9e556afd44213b6bd1be2b850ebbbd98f5481437a8021afaf58ee7fb1818d347
3
+ size 499723
tokenizer_config.json ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_bos_token": true,
3
+ "add_eos_token": false,
4
+ "add_prefix_space": null,
5
+ "added_tokens_decoder": {
6
+ "0": {
7
+ "content": "<unk>",
8
+ "lstrip": false,
9
+ "normalized": false,
10
+ "rstrip": false,
11
+ "single_word": false,
12
+ "special": true
13
+ },
14
+ "1": {
15
+ "content": "<s>",
16
+ "lstrip": false,
17
+ "normalized": false,
18
+ "rstrip": false,
19
+ "single_word": false,
20
+ "special": true
21
+ },
22
+ "2": {
23
+ "content": "</s>",
24
+ "lstrip": false,
25
+ "normalized": false,
26
+ "rstrip": false,
27
+ "single_word": false,
28
+ "special": true
29
+ },
30
+ "32000": {
31
+ "content": "<|end_of_text|>",
32
+ "lstrip": false,
33
+ "normalized": false,
34
+ "rstrip": false,
35
+ "single_word": false,
36
+ "special": true
37
+ }
38
+ },
39
+ "bos_token": "<s>",
40
+ "chat_template": "{% for message in messages %}\n{% if message['role'] == 'user' %}\n{{ '<|user|>\n' + message['content'] + eos_token }}\n{% elif message['role'] == 'system' %}\n{{ '<|system|>\n' + message['content'] + eos_token }}\n{% elif message['role'] == 'assistant' %}\n{{ '<|assistant|>\n' + message['content'] + eos_token }}\n{% endif %}\n{% if loop.last and add_generation_prompt %}\n{{ '<|assistant|>' }}\n{% endif %}\n{% endfor %}",
41
+ "clean_up_tokenization_spaces": false,
42
+ "eos_token": "</s>",
43
+ "extra_special_tokens": {},
44
+ "legacy": false,
45
+ "model_max_length": 2048,
46
+ "pad_token": "<|end_of_text|>",
47
+ "padding_side": "right",
48
+ "sp_model_kwargs": {},
49
+ "tokenizer_class": "LlamaTokenizer",
50
+ "unk_token": "<unk>",
51
+ "use_default_system_prompt": false
52
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b7ec7cf72bae970f348408b8665cfad5c138e242b2f79cd44b5ea518c4829599
3
+ size 6328