JIN223 commited on
Commit
ef33f6b
·
verified ·
1 Parent(s): 7678070

Upload folder using huggingface_hub

Browse files
README.md ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: keras-hub
3
+ pipeline_tag: text-generation
4
+ ---
5
+ This is a [`GPT2` model](https://keras.io/api/keras_hub/models/gpt2) uploaded using the KerasHub library and can be used with JAX, TensorFlow, and PyTorch backends.
6
+ This model is related to a `CausalLM` task.
7
+
8
+ Model config:
9
+ * **name:** gpt2_backbone
10
+ * **trainable:** True
11
+ * **vocabulary_size:** 50257
12
+ * **num_layers:** 12
13
+ * **num_heads:** 12
14
+ * **hidden_dim:** 768
15
+ * **intermediate_dim:** 3072
16
+ * **dropout:** 0.1
17
+ * **max_sequence_length:** 1024
18
+
19
+ This model card has been generated automatically and should be completed by the model author. See [Model Cards documentation](https://huggingface.co/docs/hub/model-cards) for more information.
assets/tokenizer/merges.txt ADDED
The diff for this file is too large to render. See raw diff
 
assets/tokenizer/vocabulary.json ADDED
The diff for this file is too large to render. See raw diff
 
config.json ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "module": "keras_hub.src.models.gpt2.gpt2_backbone",
3
+ "class_name": "GPT2Backbone",
4
+ "config": {
5
+ "name": "gpt2_backbone",
6
+ "trainable": true,
7
+ "vocabulary_size": 50257,
8
+ "num_layers": 12,
9
+ "num_heads": 12,
10
+ "hidden_dim": 768,
11
+ "intermediate_dim": 3072,
12
+ "dropout": 0.1,
13
+ "max_sequence_length": 1024
14
+ },
15
+ "registered_name": "keras_hub>GPT2Backbone"
16
+ }
metadata.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "keras_version": "3.8.0",
3
+ "keras_hub_version": "0.19.3",
4
+ "parameter_count": 124439808,
5
+ "date_saved": "2025-03-31@01:09:23",
6
+ "tasks": [
7
+ "CausalLM"
8
+ ]
9
+ }
model.weights.h5 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e09148ceee1b8d2edb3aacd7555af35d26830b17bd6a45dbea26ff4686acef46
3
+ size 498184528
preprocessor.json ADDED
@@ -0,0 +1,44 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "module": "keras_hub.src.models.gpt2.gpt2_causal_lm_preprocessor",
3
+ "class_name": "GPT2CausalLMPreprocessor",
4
+ "config": {
5
+ "name": "gpt2_causal_lm_preprocessor",
6
+ "trainable": true,
7
+ "dtype": {
8
+ "module": "keras",
9
+ "class_name": "DTypePolicy",
10
+ "config": {
11
+ "name": "float32"
12
+ },
13
+ "registered_name": null
14
+ },
15
+ "tokenizer": {
16
+ "module": "keras_hub.src.models.gpt2.gpt2_tokenizer",
17
+ "class_name": "GPT2Tokenizer",
18
+ "config": {
19
+ "name": "gpt2_tokenizer",
20
+ "trainable": true,
21
+ "dtype": {
22
+ "module": "keras",
23
+ "class_name": "DTypePolicy",
24
+ "config": {
25
+ "name": "int32"
26
+ },
27
+ "registered_name": null
28
+ },
29
+ "config_file": "tokenizer.json",
30
+ "sequence_length": null,
31
+ "add_prefix_space": false,
32
+ "unsplittable_tokens": [
33
+ "<|endoftext|>"
34
+ ]
35
+ },
36
+ "registered_name": "keras_hub>GPT2Tokenizer"
37
+ },
38
+ "config_file": "preprocessor.json",
39
+ "sequence_length": 1024,
40
+ "add_start_token": true,
41
+ "add_end_token": true
42
+ },
43
+ "registered_name": "keras_hub>GPT2CausalLMPreprocessor"
44
+ }
task.json ADDED
@@ -0,0 +1,68 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "module": "keras_hub.src.models.gpt2.gpt2_causal_lm",
3
+ "class_name": "GPT2CausalLM",
4
+ "config": {
5
+ "backbone": {
6
+ "module": "keras_hub.src.models.gpt2.gpt2_backbone",
7
+ "class_name": "GPT2Backbone",
8
+ "config": {
9
+ "name": "gpt2_backbone",
10
+ "trainable": true,
11
+ "vocabulary_size": 50257,
12
+ "num_layers": 12,
13
+ "num_heads": 12,
14
+ "hidden_dim": 768,
15
+ "intermediate_dim": 3072,
16
+ "dropout": 0.1,
17
+ "max_sequence_length": 1024
18
+ },
19
+ "registered_name": "keras_hub>GPT2Backbone"
20
+ },
21
+ "preprocessor": {
22
+ "module": "keras_hub.src.models.gpt2.gpt2_causal_lm_preprocessor",
23
+ "class_name": "GPT2CausalLMPreprocessor",
24
+ "config": {
25
+ "name": "gpt2_causal_lm_preprocessor",
26
+ "trainable": true,
27
+ "dtype": {
28
+ "module": "keras",
29
+ "class_name": "DTypePolicy",
30
+ "config": {
31
+ "name": "float32"
32
+ },
33
+ "registered_name": null
34
+ },
35
+ "tokenizer": {
36
+ "module": "keras_hub.src.models.gpt2.gpt2_tokenizer",
37
+ "class_name": "GPT2Tokenizer",
38
+ "config": {
39
+ "name": "gpt2_tokenizer",
40
+ "trainable": true,
41
+ "dtype": {
42
+ "module": "keras",
43
+ "class_name": "DTypePolicy",
44
+ "config": {
45
+ "name": "int32"
46
+ },
47
+ "registered_name": null
48
+ },
49
+ "config_file": "tokenizer.json",
50
+ "sequence_length": null,
51
+ "add_prefix_space": false,
52
+ "unsplittable_tokens": [
53
+ "<|endoftext|>"
54
+ ]
55
+ },
56
+ "registered_name": "keras_hub>GPT2Tokenizer"
57
+ },
58
+ "config_file": "preprocessor.json",
59
+ "sequence_length": 1024,
60
+ "add_start_token": true,
61
+ "add_end_token": true
62
+ },
63
+ "registered_name": "keras_hub>GPT2CausalLMPreprocessor"
64
+ },
65
+ "name": "gpt2_causal_lm"
66
+ },
67
+ "registered_name": "keras_hub>GPT2CausalLM"
68
+ }
tokenizer.json ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "module": "keras_hub.src.models.gpt2.gpt2_tokenizer",
3
+ "class_name": "GPT2Tokenizer",
4
+ "config": {
5
+ "name": "gpt2_tokenizer",
6
+ "trainable": true,
7
+ "dtype": {
8
+ "module": "keras",
9
+ "class_name": "DTypePolicy",
10
+ "config": {
11
+ "name": "int32"
12
+ },
13
+ "registered_name": null
14
+ },
15
+ "config_file": "tokenizer.json",
16
+ "sequence_length": null,
17
+ "add_prefix_space": false,
18
+ "unsplittable_tokens": [
19
+ "<|endoftext|>"
20
+ ]
21
+ },
22
+ "registered_name": "keras_hub>GPT2Tokenizer"
23
+ }