tameto commited on
Commit
ce15747
·
verified ·
1 Parent(s): f1347ea

Initial kimetsu-style tiny GPT

Browse files
README.md CHANGED
@@ -13,8 +13,8 @@ tags:
13
 
14
  # Kimetsu-Style GPT (Tiny)
15
 
16
- **“鬼滅風”のセリフを生成する超小型デモモデル**(教育/勉強会向け)。
17
- このノートで、**自作コーパス → トークナイズ → 事前学習 → 生成 → 公開**までを約1時間で再現しています。
18
 
19
  ## Training
20
  - Tokenizer: ByteLevel BPE (~5000)
 
13
 
14
  # Kimetsu-Style GPT (Tiny)
15
 
16
+ **“鬼滅風”のセリフを生成する超小型デモモデル**(教育/勉強会向け)。
17
+ このノートで、**自作コーパス → トークナイズ → 事前学習 → 生成 → 公開**まで実施しています。
18
 
19
  ## Training
20
  - Tokenizer: ByteLevel BPE (~5000)
config.json CHANGED
@@ -4,7 +4,7 @@
4
  "GPT2LMHeadModel"
5
  ],
6
  "attn_pdrop": 0.1,
7
- "bos_token_id": 0,
8
  "dtype": "float32",
9
  "embd_pdrop": 0.1,
10
  "eos_token_id": 2,
@@ -28,5 +28,5 @@
28
  "summary_use_proj": true,
29
  "transformers_version": "4.56.1",
30
  "use_cache": true,
31
- "vocab_size": 412
32
  }
 
4
  "GPT2LMHeadModel"
5
  ],
6
  "attn_pdrop": 0.1,
7
+ "bos_token_id": 1,
8
  "dtype": "float32",
9
  "embd_pdrop": 0.1,
10
  "eos_token_id": 2,
 
28
  "summary_use_proj": true,
29
  "transformers_version": "4.56.1",
30
  "use_cache": true,
31
+ "vocab_size": 32000
32
  }
generation_config.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "_from_model_config": true,
3
- "bos_token_id": 0,
4
  "eos_token_id": 2,
5
  "transformers_version": "4.56.1"
6
  }
 
1
  {
2
  "_from_model_config": true,
3
+ "bos_token_id": 1,
4
  "eos_token_id": 2,
5
  "transformers_version": "4.56.1"
6
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c0afc1c59b629af6065545d201338189d736b716f1a78d5c878209a8abad3c3f
3
- size 6875816
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:504d915f248c64adad7ad6677af48d648944d1d93c33fb4e44c650a1a5e3aed8
3
+ size 39221936
special_tokens_map.json CHANGED
@@ -1,11 +1,6 @@
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,
@@ -13,13 +8,15 @@
13
  "rstrip": false,
14
  "single_word": false
15
  },
 
16
  "pad_token": {
17
- "content": "<pad>",
18
  "lstrip": false,
19
  "normalized": false,
20
  "rstrip": false,
21
  "single_word": false
22
  },
 
23
  "unk_token": {
24
  "content": "<unk>",
25
  "lstrip": false,
 
1
  {
2
+ "bos_token": "<s>",
3
+ "cls_token": "[CLS]",
 
 
 
 
 
4
  "eos_token": {
5
  "content": "</s>",
6
  "lstrip": false,
 
8
  "rstrip": false,
9
  "single_word": false
10
  },
11
+ "mask_token": "[MASK]",
12
  "pad_token": {
13
+ "content": "[PAD]",
14
  "lstrip": false,
15
  "normalized": false,
16
  "rstrip": false,
17
  "single_word": false
18
  },
19
+ "sep_token": "[SEP]",
20
  "unk_token": {
21
  "content": "<unk>",
22
  "lstrip": false,
spiece.model ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b5cbdfa8aa7c54c8c5af85b78c309c54a5f2749a20468bf6f60eee007fe6fec1
3
+ size 805634
tokenizer.json CHANGED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json CHANGED
@@ -1,7 +1,8 @@
1
  {
 
2
  "added_tokens_decoder": {
3
  "0": {
4
- "content": "<s>",
5
  "lstrip": false,
6
  "normalized": false,
7
  "rstrip": false,
@@ -9,7 +10,7 @@
9
  "special": true
10
  },
11
  "1": {
12
- "content": "<pad>",
13
  "lstrip": false,
14
  "normalized": false,
15
  "rstrip": false,
@@ -25,7 +26,31 @@
25
  "special": true
26
  },
27
  "3": {
28
- "content": "<unk>",
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
29
  "lstrip": false,
30
  "normalized": false,
31
  "rstrip": false,
@@ -33,12 +58,20 @@
33
  "special": true
34
  }
35
  },
 
36
  "bos_token": "<s>",
37
  "clean_up_tokenization_spaces": false,
 
 
38
  "eos_token": "</s>",
 
39
  "extra_special_tokens": {},
 
 
40
  "model_max_length": 1000000000000000019884624838656,
41
- "pad_token": "<pad>",
42
- "tokenizer_class": "PreTrainedTokenizerFast",
 
 
43
  "unk_token": "<unk>"
44
  }
 
1
  {
2
+ "add_prefix_space": null,
3
  "added_tokens_decoder": {
4
  "0": {
5
+ "content": "<unk>",
6
  "lstrip": false,
7
  "normalized": false,
8
  "rstrip": false,
 
10
  "special": true
11
  },
12
  "1": {
13
+ "content": "<s>",
14
  "lstrip": false,
15
  "normalized": false,
16
  "rstrip": false,
 
26
  "special": true
27
  },
28
  "3": {
29
+ "content": "[PAD]",
30
+ "lstrip": false,
31
+ "normalized": false,
32
+ "rstrip": false,
33
+ "single_word": false,
34
+ "special": true
35
+ },
36
+ "4": {
37
+ "content": "[CLS]",
38
+ "lstrip": false,
39
+ "normalized": false,
40
+ "rstrip": false,
41
+ "single_word": false,
42
+ "special": true
43
+ },
44
+ "5": {
45
+ "content": "[SEP]",
46
+ "lstrip": false,
47
+ "normalized": false,
48
+ "rstrip": false,
49
+ "single_word": false,
50
+ "special": true
51
+ },
52
+ "6": {
53
+ "content": "[MASK]",
54
  "lstrip": false,
55
  "normalized": false,
56
  "rstrip": false,
 
58
  "special": true
59
  }
60
  },
61
+ "additional_special_tokens": [],
62
  "bos_token": "<s>",
63
  "clean_up_tokenization_spaces": false,
64
+ "cls_token": "[CLS]",
65
+ "do_lower_case": true,
66
  "eos_token": "</s>",
67
+ "extra_ids": 0,
68
  "extra_special_tokens": {},
69
+ "legacy": true,
70
+ "mask_token": "[MASK]",
71
  "model_max_length": 1000000000000000019884624838656,
72
+ "pad_token": "[PAD]",
73
+ "sep_token": "[SEP]",
74
+ "sp_model_kwargs": {},
75
+ "tokenizer_class": "T5Tokenizer",
76
  "unk_token": "<unk>"
77
  }
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:7b761e97981f33bde7c70dfe5c9b438eac49034f1648cbdce0a9a315b71b86e5
3
  size 5777
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:83477d2c8dfea6010fdbc937e2b76bfb8940d94a0bd43cdb4777bf525745985d
3
  size 5777