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

Super-squash branch 'main' using huggingface_hub

Browse files
Files changed (5) hide show
  1. .gitattributes +36 -0
  2. README.md +77 -0
  3. kf_config.json +23 -0
  4. model.weights.h5 +3 -0
  5. tokenizer.json +3 -0
.gitattributes ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
36
+ tokenizer.json filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,77 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ pipeline_tag: text-generation
3
+ license: gemma
4
+ base_model: google/gemma-2b-it
5
+ library_name: kerasformers
6
+ extra_gated_heading: Access Gemma on Hugging Face
7
+ language:
8
+ - en
9
+ tags:
10
+ - keras
11
+ - kerasformers
12
+ - gemma
13
+ - gemma-2b
14
+ - text-generation
15
+ - arxiv:2403.08295
16
+ - pytorch
17
+ - jax
18
+ - tf
19
+ ---
20
+
21
+ *See [our collection](https://huggingface.co/kerasformers) for all Gemma sizes and variants.*
22
+
23
+ # Run Gemma with Keras 3: JAX, PyTorch, or TensorFlow
24
+
25
+ [![GitHub](https://img.shields.io/badge/GitHub-KerasFormers-181717?logo=github)](https://github.com/IMvision12/KerasFormers) [![Docs](https://img.shields.io/badge/Docs-Gemma-1f6feb)](https://imvision12.github.io/KerasFormers/gemma/) [![HuggingFace](https://img.shields.io/badge/HuggingFace-Gemma-ffd21e?logo=huggingface&logoColor=black)](https://huggingface.co/kerasformers)
26
+
27
+ # kerasformers/gemma-2b-it
28
+
29
+ Pure-**Keras 3** conversion of [`google/gemma-2b-it`](https://huggingface.co/google/gemma-2b-it) for
30
+ [kerasformers](https://github.com/IMvision12/KerasFormers). One implementation runs unmodified on
31
+ **TensorFlow / Torch / JAX**. This is the instruction-tuned checkpoint, served here as **text -> text** via `GemmaTextGenerate`; weights are
32
+ stored in **bfloat16**.
33
+
34
+ For model details, license, and usage terms, see Google's
35
+ [model card](https://huggingface.co/google/gemma-2b-it).
36
+
37
+ ## ✨ Quick start
38
+
39
+ ```python
40
+ import os
41
+ os.environ["KERAS_BACKEND"] = "torch" # or "jax" / "tensorflow"
42
+
43
+ from kerasformers.models.gemma import GemmaTextGenerate, GemmaTokenizer
44
+
45
+ model = GemmaTextGenerate.from_weights("kerasformers/gemma-2b-it")
46
+ tokenizer = GemmaTokenizer.from_weights("kerasformers/gemma-2b-it")
47
+
48
+ inputs = tokenizer([{"role": "user", "content": "Hello, who are you?"}])
49
+ outputs = model.generate(**inputs, max_new_tokens=64)
50
+ print(tokenizer.decode(outputs[0]))
51
+ ```
52
+
53
+ Load any Gemma variant the same way with `from_weights("kerasformers/<variant>")`:
54
+
55
+ | Variant | Hub |
56
+ | --- | --- |
57
+ | `gemma-1.1-2b-it` | [kerasformers/gemma-1.1-2b-it](https://huggingface.co/kerasformers/gemma-1.1-2b-it) |
58
+ | `gemma-1.1-7b-it` | [kerasformers/gemma-1.1-7b-it](https://huggingface.co/kerasformers/gemma-1.1-7b-it) |
59
+ | `gemma-2b` | [kerasformers/gemma-2b](https://huggingface.co/kerasformers/gemma-2b) |
60
+ | `gemma-2b-it` | [kerasformers/gemma-2b-it](https://huggingface.co/kerasformers/gemma-2b-it) |
61
+ | `gemma-7b` | [kerasformers/gemma-7b](https://huggingface.co/kerasformers/gemma-7b) |
62
+ | `gemma-7b-it` | [kerasformers/gemma-7b-it](https://huggingface.co/kerasformers/gemma-7b-it) |
63
+
64
+ ## Tips
65
+
66
+ - Set `KERAS_BACKEND` **before** importing Keras / kerasformers.
67
+ - Loads in **bfloat16** by default. Pass `load_dtype="float32"` for full precision,
68
+ or `quantization="int8"` to shrink further.
69
+ - See the [Gemma docs](https://imvision12.github.io/KerasFormers/gemma/).
70
+ - Community / upstream weights still work via the `hf:` prefix:
71
+ `GemmaTextGenerate.from_weights("hf:google/gemma-2b-it")`.
72
+
73
+ ## Special Thanks
74
+
75
+ A huge thank you to the Google Gemma authors for creating and releasing these models.
76
+
77
+ License: Gemma (gated). Accept the license on the upstream Hub card before downloading.
kf_config.json ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "library_name": "kerasformers",
3
+ "kerasformers_version": "1.2.1",
4
+ "model_module": "kerasformers.models.gemma",
5
+ "model_class": "GemmaTextGenerate",
6
+ "variant": "gemma-2b-it",
7
+ "weights": "model.weights.h5",
8
+ "schema_version": 2,
9
+ "weight_dtype": "bfloat16",
10
+ "model_type": "gemma",
11
+ "text_config": {
12
+ "vocab_size": 256000,
13
+ "embed_dim": 2048,
14
+ "mlp_dim": 16384,
15
+ "num_layers": 18,
16
+ "num_heads": 8,
17
+ "num_kv_heads": 1,
18
+ "head_dim": 256,
19
+ "norm_eps": 1e-06,
20
+ "rope_theta": 10000.0,
21
+ "tie_embeddings": true
22
+ }
23
+ }
model.weights.h5 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5dfc770e2b9d51450db9137fce97dd26ab4b0e8e11a0421cf8e47864dc62d5ac
3
+ size 5013657408
tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c15eb04bc5ad609fb26533e8525302c5640a945e5f67f65b7c849900acda7d99
3
+ size 17518497