Allen Poston commited on
Commit
e6a2c37
·
verified ·
1 Parent(s): 65da3e5

Model save

Browse files
README.md CHANGED
@@ -1,52 +1,53 @@
1
- ---
2
- library_name: transformers
3
- license: mit
4
- base_model: microsoft/DialoGPT-medium
5
- tags:
6
- - generated_from_trainer
7
- model-index:
8
- - name: anime-senko-chat
9
- results: []
10
- ---
11
-
12
- <!-- This model card has been generated automatically according to the information the Trainer had access to. You
13
- should probably proofread and complete it, then remove this comment. -->
14
-
15
- # anime-senko-chat
16
-
17
- This model is a fine-tuned version of [microsoft/DialoGPT-medium](https://huggingface.co/microsoft/DialoGPT-medium) on the None dataset.
18
-
19
- ## Model description
20
-
21
- More information needed
22
-
23
- ## Intended uses & limitations
24
-
25
- More information needed
26
-
27
- ## Training and evaluation data
28
-
29
- More information needed
30
-
31
- ## Training procedure
32
-
33
- ### Training hyperparameters
34
-
35
- The following hyperparameters were used during training:
36
- - learning_rate: 5e-05
37
- - train_batch_size: 8
38
- - eval_batch_size: 8
39
- - seed: 42
40
- - gradient_accumulation_steps: 4
41
- - total_train_batch_size: 32
42
- - optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
43
- - lr_scheduler_type: cosine
44
- - lr_scheduler_warmup_steps: 50
45
- - num_epochs: 2
46
-
47
- ### Framework versions
48
-
49
- - Transformers 4.50.0
50
- - Pytorch 2.5.1+cpu
51
- - Datasets 3.6.0
52
- - Tokenizers 0.21.1
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: mit
4
+ base_model: microsoft/DialoGPT-medium
5
+ tags:
6
+ - generated_from_trainer
7
+ model-index:
8
+ - name: anime-senko-chat
9
+ results: []
10
+ ---
11
+
12
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
13
+ should probably proofread and complete it, then remove this comment. -->
14
+
15
+ # anime-senko-chat
16
+
17
+ This model is a fine-tuned version of [microsoft/DialoGPT-medium](https://huggingface.co/microsoft/DialoGPT-medium) on the None dataset.
18
+
19
+ ## Model description
20
+
21
+ More information needed
22
+
23
+ ## Intended uses & limitations
24
+
25
+ More information needed
26
+
27
+ ## Training and evaluation data
28
+
29
+ More information needed
30
+
31
+ ## Training procedure
32
+
33
+ ### Training hyperparameters
34
+
35
+ The following hyperparameters were used during training:
36
+ - learning_rate: 5e-05
37
+ - train_batch_size: 8
38
+ - eval_batch_size: 8
39
+ - seed: 42
40
+ - gradient_accumulation_steps: 4
41
+ - total_train_batch_size: 32
42
+ - optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
43
+ - lr_scheduler_type: linear
44
+ - lr_scheduler_warmup_steps: 50
45
+ - num_epochs: 2
46
+ - mixed_precision_training: Native AMP
47
+
48
+ ### Framework versions
49
+
50
+ - Transformers 4.51.3
51
+ - Pytorch 2.6.0+cu124
52
+ - Datasets 2.14.4
53
+ - Tokenizers 0.21.1
added_tokens.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "<|assistant|>": 50258,
3
+ "<|context|>": 50260,
4
+ "<|human|>": 50257,
5
+ "<|system|>": 50259
6
+ }
config.json CHANGED
@@ -1,37 +1,37 @@
1
- {
2
- "activation_function": "gelu_new",
3
- "architectures": [
4
- "GPT2LMHeadModel"
5
- ],
6
- "attn_pdrop": 0.1,
7
- "bos_token_id": 50256,
8
- "embd_pdrop": 0.1,
9
- "eos_token_id": 50256,
10
- "initializer_range": 0.02,
11
- "layer_norm_epsilon": 1e-05,
12
- "model_type": "gpt2",
13
- "n_ctx": 1024,
14
- "n_embd": 1024,
15
- "n_head": 16,
16
- "n_inner": null,
17
- "n_layer": 24,
18
- "n_positions": 1024,
19
- "reorder_and_upcast_attn": false,
20
- "resid_pdrop": 0.1,
21
- "scale_attn_by_inverse_layer_idx": false,
22
- "scale_attn_weights": true,
23
- "summary_activation": null,
24
- "summary_first_dropout": 0.1,
25
- "summary_proj_to_labels": true,
26
- "summary_type": "cls_index",
27
- "summary_use_proj": true,
28
- "task_specific_params": {
29
- "conversational": {
30
- "max_length": 1000
31
- }
32
- },
33
- "torch_dtype": "float32",
34
- "transformers_version": "4.50.0",
35
- "use_cache": false,
36
- "vocab_size": 50257
37
- }
 
1
+ {
2
+ "activation_function": "gelu_new",
3
+ "architectures": [
4
+ "GPT2LMHeadModel"
5
+ ],
6
+ "attn_pdrop": 0.1,
7
+ "bos_token_id": 50256,
8
+ "embd_pdrop": 0.1,
9
+ "eos_token_id": 50256,
10
+ "initializer_range": 0.02,
11
+ "layer_norm_epsilon": 1e-05,
12
+ "model_type": "gpt2",
13
+ "n_ctx": 1024,
14
+ "n_embd": 1024,
15
+ "n_head": 16,
16
+ "n_inner": null,
17
+ "n_layer": 24,
18
+ "n_positions": 1024,
19
+ "reorder_and_upcast_attn": false,
20
+ "resid_pdrop": 0.1,
21
+ "scale_attn_by_inverse_layer_idx": false,
22
+ "scale_attn_weights": true,
23
+ "summary_activation": null,
24
+ "summary_first_dropout": 0.1,
25
+ "summary_proj_to_labels": true,
26
+ "summary_type": "cls_index",
27
+ "summary_use_proj": true,
28
+ "task_specific_params": {
29
+ "conversational": {
30
+ "max_length": 1000
31
+ }
32
+ },
33
+ "torch_dtype": "float16",
34
+ "transformers_version": "4.51.3",
35
+ "use_cache": false,
36
+ "vocab_size": 50261
37
+ }
generation_config.json CHANGED
@@ -1,6 +1,6 @@
1
- {
2
- "_from_model_config": true,
3
- "bos_token_id": 50256,
4
- "eos_token_id": 50256,
5
- "transformers_version": "4.50.0"
6
- }
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bos_token_id": 50256,
4
+ "eos_token_id": 50256,
5
+ "transformers_version": "4.51.3"
6
+ }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:65d091f5d3687b3258a7489b869ec1bc0fa403da80a27d1b00c37db9520d9f8f
3
- size 1419322880
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:997cfed25f9f5350164febeea24bac38ae45a6cfd366c55186171b057afa71e5
3
+ size 709684568
special_tokens_map.json CHANGED
@@ -1,24 +1,54 @@
1
- {
2
- "bos_token": {
3
- "content": "<|endoftext|>",
4
- "lstrip": false,
5
- "normalized": true,
6
- "rstrip": false,
7
- "single_word": false
8
- },
9
- "eos_token": {
10
- "content": "<|endoftext|>",
11
- "lstrip": false,
12
- "normalized": true,
13
- "rstrip": false,
14
- "single_word": false
15
- },
16
- "pad_token": "<|endoftext|>",
17
- "unk_token": {
18
- "content": "<|endoftext|>",
19
- "lstrip": false,
20
- "normalized": true,
21
- "rstrip": false,
22
- "single_word": false
23
- }
24
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "additional_special_tokens": [
3
+ {
4
+ "content": "<|human|>",
5
+ "lstrip": false,
6
+ "normalized": false,
7
+ "rstrip": false,
8
+ "single_word": false
9
+ },
10
+ {
11
+ "content": "<|assistant|>",
12
+ "lstrip": false,
13
+ "normalized": false,
14
+ "rstrip": false,
15
+ "single_word": false
16
+ },
17
+ {
18
+ "content": "<|system|>",
19
+ "lstrip": false,
20
+ "normalized": false,
21
+ "rstrip": false,
22
+ "single_word": false
23
+ },
24
+ {
25
+ "content": "<|context|>",
26
+ "lstrip": false,
27
+ "normalized": false,
28
+ "rstrip": false,
29
+ "single_word": false
30
+ }
31
+ ],
32
+ "bos_token": {
33
+ "content": "<|endoftext|>",
34
+ "lstrip": false,
35
+ "normalized": true,
36
+ "rstrip": false,
37
+ "single_word": false
38
+ },
39
+ "eos_token": {
40
+ "content": "<|endoftext|>",
41
+ "lstrip": false,
42
+ "normalized": true,
43
+ "rstrip": false,
44
+ "single_word": false
45
+ },
46
+ "pad_token": "<|endoftext|>",
47
+ "unk_token": {
48
+ "content": "<|endoftext|>",
49
+ "lstrip": false,
50
+ "normalized": true,
51
+ "rstrip": false,
52
+ "single_word": false
53
+ }
54
+ }
tokenizer.json CHANGED
@@ -25,6 +25,42 @@
25
  "rstrip": false,
26
  "normalized": true,
27
  "special": true
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
28
  }
29
  ],
30
  "normalizer": null,
 
25
  "rstrip": false,
26
  "normalized": true,
27
  "special": true
28
+ },
29
+ {
30
+ "id": 50257,
31
+ "content": "<|human|>",
32
+ "single_word": false,
33
+ "lstrip": false,
34
+ "rstrip": false,
35
+ "normalized": false,
36
+ "special": true
37
+ },
38
+ {
39
+ "id": 50258,
40
+ "content": "<|assistant|>",
41
+ "single_word": false,
42
+ "lstrip": false,
43
+ "rstrip": false,
44
+ "normalized": false,
45
+ "special": true
46
+ },
47
+ {
48
+ "id": 50259,
49
+ "content": "<|system|>",
50
+ "single_word": false,
51
+ "lstrip": false,
52
+ "rstrip": false,
53
+ "normalized": false,
54
+ "special": true
55
+ },
56
+ {
57
+ "id": 50260,
58
+ "content": "<|context|>",
59
+ "single_word": false,
60
+ "lstrip": false,
61
+ "rstrip": false,
62
+ "normalized": false,
63
+ "special": true
64
  }
65
  ],
66
  "normalizer": null,
tokenizer_config.json CHANGED
@@ -1,24 +1,62 @@
1
- {
2
- "add_bos_token": false,
3
- "add_prefix_space": false,
4
- "added_tokens_decoder": {
5
- "50256": {
6
- "content": "<|endoftext|>",
7
- "lstrip": false,
8
- "normalized": true,
9
- "rstrip": false,
10
- "single_word": false,
11
- "special": true
12
- }
13
- },
14
- "bos_token": "<|endoftext|>",
15
- "chat_template": "{% for message in messages %}{{ message.content }}{{ eos_token }}{% endfor %}",
16
- "clean_up_tokenization_spaces": true,
17
- "eos_token": "<|endoftext|>",
18
- "errors": "replace",
19
- "extra_special_tokens": {},
20
- "model_max_length": 1024,
21
- "pad_token": "<|endoftext|>",
22
- "tokenizer_class": "GPT2Tokenizer",
23
- "unk_token": "<|endoftext|>"
24
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_bos_token": false,
3
+ "add_prefix_space": false,
4
+ "added_tokens_decoder": {
5
+ "50256": {
6
+ "content": "<|endoftext|>",
7
+ "lstrip": false,
8
+ "normalized": true,
9
+ "rstrip": false,
10
+ "single_word": false,
11
+ "special": true
12
+ },
13
+ "50257": {
14
+ "content": "<|human|>",
15
+ "lstrip": false,
16
+ "normalized": false,
17
+ "rstrip": false,
18
+ "single_word": false,
19
+ "special": true
20
+ },
21
+ "50258": {
22
+ "content": "<|assistant|>",
23
+ "lstrip": false,
24
+ "normalized": false,
25
+ "rstrip": false,
26
+ "single_word": false,
27
+ "special": true
28
+ },
29
+ "50259": {
30
+ "content": "<|system|>",
31
+ "lstrip": false,
32
+ "normalized": false,
33
+ "rstrip": false,
34
+ "single_word": false,
35
+ "special": true
36
+ },
37
+ "50260": {
38
+ "content": "<|context|>",
39
+ "lstrip": false,
40
+ "normalized": false,
41
+ "rstrip": false,
42
+ "single_word": false,
43
+ "special": true
44
+ }
45
+ },
46
+ "additional_special_tokens": [
47
+ "<|human|>",
48
+ "<|assistant|>",
49
+ "<|system|>",
50
+ "<|context|>"
51
+ ],
52
+ "bos_token": "<|endoftext|>",
53
+ "chat_template": "{% for message in messages %}{{ message.content }}{{ eos_token }}{% endfor %}",
54
+ "clean_up_tokenization_spaces": true,
55
+ "eos_token": "<|endoftext|>",
56
+ "errors": "replace",
57
+ "extra_special_tokens": {},
58
+ "model_max_length": 1024,
59
+ "pad_token": "<|endoftext|>",
60
+ "tokenizer_class": "GPT2Tokenizer",
61
+ "unk_token": "<|endoftext|>"
62
+ }
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c213159366931a5cdcf52e959abd202cd48a7caa53b70b4b024f34ae32854855
3
- size 5368
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:03eea7b7b1a2324a40b9e3e48103eae1661deb39e8de6eecbf835f4d7cb7101b
3
+ size 5304