IMvision12 commited on
Commit
c7c9643
·
0 Parent(s):

Super-squash branch 'main' using huggingface_hub

Browse files
Files changed (5) hide show
  1. .gitattributes +35 -0
  2. README.md +76 -0
  3. kf_config.json +21 -0
  4. model.weights.h5 +3 -0
  5. tokenizer.json +0 -0
.gitattributes ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ *.7z filter=lfs diff=lfs merge=lfs -text
2
+ *.arrow filter=lfs diff=lfs merge=lfs -text
3
+ *.bin filter=lfs diff=lfs merge=lfs -text
4
+ *.bz2 filter=lfs diff=lfs merge=lfs -text
5
+ *.ckpt filter=lfs diff=lfs merge=lfs -text
6
+ *.ftz filter=lfs diff=lfs merge=lfs -text
7
+ *.gz filter=lfs diff=lfs merge=lfs -text
8
+ *.h5 filter=lfs diff=lfs merge=lfs -text
9
+ *.joblib filter=lfs diff=lfs merge=lfs -text
10
+ *.lfs.* filter=lfs diff=lfs merge=lfs -text
11
+ *.mlmodel filter=lfs diff=lfs merge=lfs -text
12
+ *.model filter=lfs diff=lfs merge=lfs -text
13
+ *.msgpack filter=lfs diff=lfs merge=lfs -text
14
+ *.npy filter=lfs diff=lfs merge=lfs -text
15
+ *.npz filter=lfs diff=lfs merge=lfs -text
16
+ *.onnx filter=lfs diff=lfs merge=lfs -text
17
+ *.ot filter=lfs diff=lfs merge=lfs -text
18
+ *.parquet filter=lfs diff=lfs merge=lfs -text
19
+ *.pb filter=lfs diff=lfs merge=lfs -text
20
+ *.pickle filter=lfs diff=lfs merge=lfs -text
21
+ *.pkl filter=lfs diff=lfs merge=lfs -text
22
+ *.pt filter=lfs diff=lfs merge=lfs -text
23
+ *.pth filter=lfs diff=lfs merge=lfs -text
24
+ *.rar filter=lfs diff=lfs merge=lfs -text
25
+ *.safetensors filter=lfs diff=lfs merge=lfs -text
26
+ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
27
+ *.tar.* filter=lfs diff=lfs merge=lfs -text
28
+ *.tar filter=lfs diff=lfs merge=lfs -text
29
+ *.tflite filter=lfs diff=lfs merge=lfs -text
30
+ *.tgz filter=lfs diff=lfs merge=lfs -text
31
+ *.wasm filter=lfs diff=lfs merge=lfs -text
32
+ *.xz filter=lfs diff=lfs merge=lfs -text
33
+ *.zip filter=lfs diff=lfs merge=lfs -text
34
+ *.zst filter=lfs diff=lfs merge=lfs -text
35
+ *tfevents* filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,76 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ pipeline_tag: text-generation
3
+ license: mit
4
+ base_model: openai-community/gpt2-large
5
+ library_name: kerasformers
6
+ language:
7
+ - en
8
+ tags:
9
+ - keras
10
+ - kerasformers
11
+ - gpt2
12
+ - gpt-2
13
+ - text-generation
14
+ - pytorch
15
+ - jax
16
+ - tf
17
+ ---
18
+
19
+ # Run GPT-2 with Keras 3: JAX, PyTorch, or TensorFlow
20
+
21
+ [![GitHub](https://img.shields.io/badge/GitHub-KerasFormers-black?logo=github)](https://github.com/IMvision12/KerasFormers) [![Docs](https://img.shields.io/badge/Docs-GPT--2-blue)](https://imvision12.github.io/KerasFormers/gpt2/)
22
+
23
+ # kerasformers/gpt2_large
24
+
25
+ Paper: [Language Models are Unsupervised Multitask Learners (Radford et al., 2019)](https://cdn.openai.com/better-language-models/language_models_are_unsupervised_multitask_learners.pdf)
26
+
27
+ GPT-2 is OpenAI's decoder-only transformer language model trained on WebText:
28
+ learned absolute position embeddings, pre-LayerNorm blocks, `gelu_new`
29
+ activations, a tied output head, and a byte-level BPE tokenizer. This is the
30
+ **774M** variant, a base completion model (no chat template).
31
+
32
+ For more details, see the upstream [model card](https://huggingface.co/openai-community/gpt2-large).
33
+
34
+ Pure-**Keras 3** conversion of [`openai-community/gpt2-large`](https://huggingface.co/openai-community/gpt2-large) for
35
+ [kerasformers](https://github.com/IMvision12/KerasFormers). One implementation runs unmodified on
36
+ **TensorFlow / Torch / JAX**.
37
+
38
+ ## Quick start
39
+
40
+ ```python
41
+ import os
42
+ os.environ["KERAS_BACKEND"] = "torch" # or "jax" / "tensorflow"
43
+
44
+ from kerasformers.models.gpt2 import GPT2TextGenerate, GPT2Tokenizer
45
+
46
+ model = GPT2TextGenerate.from_weights("kerasformers/gpt2_large")
47
+ tokenizer = GPT2Tokenizer.from_weights("kerasformers/gpt2_large")
48
+
49
+ inputs = tokenizer("The meaning of life is")
50
+ outputs = model.generate(**inputs, max_new_tokens=40)
51
+ print(tokenizer.decode(outputs[0]))
52
+ ```
53
+
54
+ All GPT-2 sizes load the same way with `from_weights("kerasformers/<variant>")`:
55
+
56
+ | Variant | Hub | Params |
57
+ |---|---|---|
58
+ | `gpt2` | [`kerasformers/gpt2`](https://huggingface.co/kerasformers/gpt2) | 124M |
59
+ | `gpt2_medium` | [`kerasformers/gpt2_medium`](https://huggingface.co/kerasformers/gpt2_medium) | 355M |
60
+ | `gpt2_large` | [`kerasformers/gpt2_large`](https://huggingface.co/kerasformers/gpt2_large) | 774M |
61
+ | `gpt2_xl` | [`kerasformers/gpt2_xl`](https://huggingface.co/kerasformers/gpt2_xl) | 1.5B |
62
+
63
+ ## Tips
64
+
65
+ - Set `KERAS_BACKEND` **before** importing Keras / kerasformers.
66
+ - This is a base completion model: it continues a prompt and is not
67
+ instruction-tuned.
68
+ - See the [GPT-2 docs](https://imvision12.github.io/KerasFormers/gpt2/) and [Loading Weights](https://imvision12.github.io/KerasFormers/loading_weights/).
69
+ - Upstream safetensors still work via the `hf:` prefix, e.g.
70
+ `GPT2TextGenerate.from_weights("hf:openai-community/gpt2-large")`.
71
+
72
+ ## Special Thanks
73
+
74
+ A huge thank you to the OpenAI GPT-2 authors for creating and releasing these models.
75
+
76
+ License: MIT, inherited from the upstream OpenAI GPT-2 release.
kf_config.json ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "library_name": "kerasformers",
3
+ "kerasformers_version": "1.2.1",
4
+ "model_module": "kerasformers.models.gpt2",
5
+ "model_class": "GPT2Model",
6
+ "variant": "gpt2_large",
7
+ "weights": "model.weights.json",
8
+ "schema_version": 2,
9
+ "weight_dtype": "float32",
10
+ "model_type": "gpt2",
11
+ "text_config": {
12
+ "vocab_size": 50257,
13
+ "embed_dim": 1280,
14
+ "mlp_dim": 5120,
15
+ "num_layers": 36,
16
+ "num_heads": 20,
17
+ "max_position_embeddings": 1024,
18
+ "norm_eps": 1e-05,
19
+ "tie_embeddings": true
20
+ }
21
+ }
model.weights.h5 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8bbcd44027fe6840d259ea8fbf11f5c6530a6b44d61d0d2dc1824558ab4fc02e
3
+ size 3096933680
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff