Upload folder using huggingface_hub

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+assets/tokenizer/vocabulary.spm filter=lfs diff=lfs merge=lfs -text

assets/tokenizer/vocabulary.spm

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:61a7b147390c64585d6c3543dd6fc636906c9af3865a5548f27f31aee1d4c8e2
+size 4241003

config.json

@@ -0,0 +1,25 @@
+{
+    "module": "keras_hub.src.models.gemma.gemma_backbone",
+    "class_name": "GemmaBackbone",
+    "config": {
+        "name": "gemma_backbone",
+        "trainable": true,
+        "vocabulary_size": 256000,
+        "num_layers": 26,
+        "num_query_heads": 8,
+        "num_key_value_heads": 4,
+        "hidden_dim": 2304,
+        "intermediate_dim": 18432,
+        "head_dim": 256,
+        "layer_norm_epsilon": 1e-06,
+        "dropout": 0,
+        "query_head_dim_normalize": true,
+        "use_post_ffw_norm": true,
+        "use_post_attention_norm": true,
+        "final_logit_soft_cap": 30.0,
+        "attention_logit_soft_cap": 50.0,
+        "sliding_window_size": 4096,
+        "use_sliding_window_attention": true
+    },
+    "registered_name": "keras_hub>GemmaBackbone"
+}

metadata.json

@@ -0,0 +1,6 @@
+{
+    "keras_version": "3.4.1",
+    "keras_hub_version": "0.18.1",
+    "parameter_count": 2626056448,
+    "date_saved": "2025-01-07@21:21:22"
+}

model.weights.h5

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b56d6fc55765e362e7bae66519f59bf0b70633eb80f71bfc413c009ef51f2463
+size 10458288832

preprocessor.json

@@ -0,0 +1,43 @@
+{
+    "module": "keras_hub.src.models.gemma.gemma_causal_lm_preprocessor",
+    "class_name": "GemmaCausalLMPreprocessor",
+    "config": {
+        "name": "gemma_causal_lm_preprocessor",
+        "trainable": true,
+        "dtype": {
+            "module": "keras",
+            "class_name": "DTypePolicy",
+            "config": {
+                "name": "float32"
+            },
+            "registered_name": null
+        },
+        "tokenizer": {
+            "module": "keras_hub.src.models.gemma.gemma_tokenizer",
+            "class_name": "GemmaTokenizer",
+            "config": {
+                "name": "gemma_tokenizer",
+                "trainable": true,
+                "dtype": {
+                    "module": "keras",
+                    "class_name": "DTypePolicy",
+                    "config": {
+                        "name": "int32"
+                    },
+                    "registered_name": null
+                },
+                "config_file": "tokenizer.json",
+                "proto": null,
+                "sequence_length": null,
+                "add_bos": false,
+                "add_eos": false
+            },
+            "registered_name": "keras_hub>GemmaTokenizer"
+        },
+        "config_file": "preprocessor.json",
+        "sequence_length": 50,
+        "add_start_token": true,
+        "add_end_token": true
+    },
+    "registered_name": "keras_hub>GemmaCausalLMPreprocessor"
+}

task.json

@@ -0,0 +1,76 @@
+{
+    "module": "keras_hub.src.models.gemma.gemma_causal_lm",
+    "class_name": "GemmaCausalLM",
+    "config": {
+        "backbone": {
+            "module": "keras_hub.src.models.gemma.gemma_backbone",
+            "class_name": "GemmaBackbone",
+            "config": {
+                "name": "gemma_backbone",
+                "trainable": true,
+                "vocabulary_size": 256000,
+                "num_layers": 26,
+                "num_query_heads": 8,
+                "num_key_value_heads": 4,
+                "hidden_dim": 2304,
+                "intermediate_dim": 18432,
+                "head_dim": 256,
+                "layer_norm_epsilon": 1e-06,
+                "dropout": 0,
+                "query_head_dim_normalize": true,
+                "use_post_ffw_norm": true,
+                "use_post_attention_norm": true,
+                "final_logit_soft_cap": 30.0,
+                "attention_logit_soft_cap": 50.0,
+                "sliding_window_size": 4096,
+                "use_sliding_window_attention": true
+            },
+            "registered_name": "keras_hub>GemmaBackbone"
+        },
+        "preprocessor": {
+            "module": "keras_hub.src.models.gemma.gemma_causal_lm_preprocessor",
+            "class_name": "GemmaCausalLMPreprocessor",
+            "config": {
+                "name": "gemma_causal_lm_preprocessor",
+                "trainable": true,
+                "dtype": {
+                    "module": "keras",
+                    "class_name": "DTypePolicy",
+                    "config": {
+                        "name": "float32"
+                    },
+                    "registered_name": null
+                },
+                "tokenizer": {
+                    "module": "keras_hub.src.models.gemma.gemma_tokenizer",
+                    "class_name": "GemmaTokenizer",
+                    "config": {
+                        "name": "gemma_tokenizer",
+                        "trainable": true,
+                        "dtype": {
+                            "module": "keras",
+                            "class_name": "DTypePolicy",
+                            "config": {
+                                "name": "int32"
+                            },
+                            "registered_name": null
+                        },
+                        "config_file": "tokenizer.json",
+                        "proto": null,
+                        "sequence_length": null,
+                        "add_bos": false,
+                        "add_eos": false
+                    },
+                    "registered_name": "keras_hub>GemmaTokenizer"
+                },
+                "config_file": "preprocessor.json",
+                "sequence_length": 50,
+                "add_start_token": true,
+                "add_end_token": true
+            },
+            "registered_name": "keras_hub>GemmaCausalLMPreprocessor"
+        },
+        "name": "gemma_causal_lm"
+    },
+    "registered_name": "keras_hub>GemmaCausalLM"
+}

tokenizer.json

@@ -0,0 +1,22 @@
+{
+    "module": "keras_hub.src.models.gemma.gemma_tokenizer",
+    "class_name": "GemmaTokenizer",
+    "config": {
+        "name": "gemma_tokenizer",
+        "trainable": true,
+        "dtype": {
+            "module": "keras",
+            "class_name": "DTypePolicy",
+            "config": {
+                "name": "int32"
+            },
+            "registered_name": null
+        },
+        "config_file": "tokenizer.json",
+        "proto": null,
+        "sequence_length": null,
+        "add_bos": false,
+        "add_eos": false
+    },
+    "registered_name": "keras_hub>GemmaTokenizer"
+}