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

Super-squash branch 'main' using huggingface_hub

Browse files
Files changed (5) hide show
  1. .gitattributes +35 -0
  2. README.md +73 -0
  3. kf_config.json +29 -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,73 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ pipeline_tag: translation
3
+ license: apache-2.0
4
+ base_model: google-t5/t5-large
5
+ library_name: kerasformers
6
+ language:
7
+ - en
8
+ - fr
9
+ - de
10
+ - ro
11
+ tags:
12
+ - keras
13
+ - kerasformers
14
+ - t5
15
+ - text2text-generation
16
+ - pytorch
17
+ - jax
18
+ - tf
19
+ ---
20
+
21
+ # T5-large in Keras 3: JAX, PyTorch, or TensorFlow
22
+
23
+ [![GitHub](https://img.shields.io/badge/GitHub-KerasFormers-181717?logo=github)](https://github.com/IMvision12/KerasFormers) [![Docs](https://img.shields.io/badge/Docs-T5-1f6feb)](https://imvision12.github.io/KerasFormers/t5/) [![HuggingFace](https://img.shields.io/badge/HuggingFace-T5-ffd21e?logo=huggingface&logoColor=black)](https://huggingface.co/collections/kerasformers/t5-6a85056935f438653698c56f)
24
+
25
+ # kerasformers/t5_large
26
+
27
+ Pure-**Keras 3** conversion of [`google-t5/t5-large`](https://huggingface.co/google-t5/t5-large) for [kerasformers](https://github.com/IMvision12/KerasFormers). One implementation runs unmodified on **TensorFlow / Torch / JAX**, bit-exact with the Hugging Face original. T5 is a text-to-text encoder-decoder; this repo hosts the full backbone (`kf_config.json` declares `T5Model`), and every T5 class (`T5ConditionalGenerate`, `T5EncoderModel`, and the classification / QA heads) loads its subset from the one `model.weights.h5`.
28
+
29
+ For model details, license, and usage terms, see the upstream [model card](https://huggingface.co/google-t5/t5-large).
30
+
31
+ Paper: [Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformer](https://arxiv.org/abs/1910.10683)
32
+
33
+ ## Quick start
34
+
35
+ ```python
36
+ import os
37
+ os.environ["KERAS_BACKEND"] = "torch" # or "jax" / "tensorflow"
38
+
39
+ from kerasformers.models.t5 import T5ConditionalGenerate, T5Tokenizer
40
+
41
+ model = T5ConditionalGenerate.from_weights("kerasformers/t5_large")
42
+ tokenizer = T5Tokenizer.from_weights("kerasformers/t5_large")
43
+
44
+ inputs = tokenizer("translate English to German: The house is wonderful.")
45
+ output_ids = model.generate(
46
+ inputs["input_ids"], inputs["attention_mask"], max_new_tokens=40
47
+ )
48
+ print(tokenizer.decode(output_ids[0]))
49
+ ```
50
+
51
+ Load any T5 variant the same way with `from_weights("kerasformers/<variant>")`. Browse them all in the [T5 collection](https://huggingface.co/collections/kerasformers/t5-6a85056935f438653698c56f).
52
+
53
+ ## Available classes
54
+
55
+ Load any of these from this repo with `from_weights("kerasformers/t5_large")` (or on the fly via the `hf:` prefix). The pretrained backbone is shared; task heads not stored in this checkpoint start randomly initialized, ready for fine-tuning (or load a `hf:` fine-tune).
56
+
57
+ | Class | Task |
58
+ |---|---|
59
+ | `T5Model` | Encoder backbone |
60
+ | `T5ConditionalGenerate` | Text-to-text generation |
61
+ | `T5EncoderModel` | Encoder-only features |
62
+ | `T5SequenceClassify` | Sequence classification |
63
+ | `T5TokenClassify` | Token classification (NER / POS) |
64
+ | `T5QnA` | Extractive question answering |
65
+
66
+ ```python
67
+ from kerasformers.models.t5 import T5SequenceClassify
68
+ model = T5SequenceClassify.from_weights("kerasformers/t5_large")
69
+ ```
70
+
71
+ ## Special Thanks
72
+
73
+ Thank you to the Google T5 team for creating and releasing the T5 models.
kf_config.json ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "library_name": "kerasformers",
3
+ "kerasformers_version": "1.2.4",
4
+ "model_module": "kerasformers.models.t5",
5
+ "model_class": "T5Model",
6
+ "variant": "t5_large",
7
+ "weights": "model.weights.h5",
8
+ "schema_version": 2,
9
+ "weight_dtype": "float32",
10
+ "model_type": "t5",
11
+ "text_config": {
12
+ "vocab_size": 32128,
13
+ "embed_dim": 1024,
14
+ "key_value_dim": 64,
15
+ "mlp_dim": 4096,
16
+ "num_layers": 24,
17
+ "num_decoder_layers": 24,
18
+ "num_heads": 16,
19
+ "relative_attention_num_buckets": 32,
20
+ "relative_attention_max_distance": 128,
21
+ "hidden_act": "relu",
22
+ "layer_norm_eps": 1e-06,
23
+ "dropout": 0.1,
24
+ "tie_word_embeddings": true,
25
+ "pad_token_id": 0,
26
+ "eos_token_id": 1,
27
+ "decoder_start_token_id": 0
28
+ }
29
+ }
model.weights.h5 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ae907456dbb174dc145c0cf2d4d01839f62fc25d1aca5aef53356a219105e9b7
3
+ size 2952483672
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff