upload model

.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
+*.spm filter=lfs diff=lfs merge=lfs -text

README.md

@@ -1,3 +1,43 @@
 ---
 license: cc-by-sa-4.0
+base_model: google/gemma-2-9b
+language:
+  - yue
+library_name: keras
+pipeline_tag: text-generation
+tags:
+  - cantonese
+  - gemma2
 ---
+
+# Cantonese LLM using Gemma-2 9B Architecture
+
+Welcome to the preview of the Cantonese Language Model (LLM) built on the Gemma-2 9B architecture. This model is designed to understand and generate text in Cantonese, including slangs, colloquials, and Internet terms.
+
+## License
+This project is available under the Creative Commons Attribution-ShareAlike 4.0 International License (CC BY-SA 4.0). For more details, please visit the [license page](https://creativecommons.org/licenses/by-sa/4.0/).
+
+## Preview Warning
+Please be advised that this version of the Cantonese LLM is a **preview**. As such, the model's outputs may sometimes be inaccurate, hallucinatory, or potentially offensive to some individuals. We are continuously working to improve the model's accuracy and reduce such instances.
+
+## Training Infrastructure
+The Cantonese LLM has been trained using Cloud TPU v3-8 VM on Google Cloud Platform. The training was done using the KerasNLP library with LoRA at learning rate of 5e-5.
+
+## Training Credits
+Google Cloud credits are provided for this project. #AISprint
+
+This model was trained by [Thomas Chong](https://huggingface.co/chongcht) and [Jacky Chan](https://huggingface.co/jhkchan) from [Votee AI Limited](https://huggingface.co/votee), and we contribute to [hon9kon9ize](https://hon9kon9ize.com/), the Hong Kong AI Research Community.
+
+## Usage Guidelines
+- Ensure that you are aware of the potential for unexpected or offensive content.
+- Always review and assess the model's output before using it in any application.
+- Provide feedback on any issues you encounter to help us improve the model.
+
+## Contributions
+We welcome contributions from the community. If you have suggestions or improvements, please submit a pull request or open an issue in the project repository.
+
+## Disclaimer
+The developers of the Cantonese LLM are not responsible for any harm or offense caused by the model's outputs. Users are advised to exercise discretion and judgment when using the model.
+
+Thank you for exploring the Cantonese LLM. We are excited to see the innovative ways in which it will be used!
+

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_nlp.src.models.gemma.gemma_backbone",
+    "class_name": "GemmaBackbone",
+    "config": {
+        "name": "gemma_backbone",
+        "trainable": true,
+        "vocabulary_size": 256000,
+        "num_layers": 42,
+        "num_query_heads": 16,
+        "num_key_value_heads": 8,
+        "hidden_dim": 3584,
+        "intermediate_dim": 28672,
+        "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,
+        "attention_logit_soft_cap": 50,
+        "sliding_window_size": 4096,
+        "use_sliding_window_attention": true
+    },
+    "registered_name": "keras_nlp>GemmaBackbone"
+}

metadata.json

@@ -0,0 +1,6 @@
+{
+    "keras_version": "3.1.1",
+    "keras_nlp_version": "0.15.1",
+    "parameter_count": 9270779392,
+    "date_saved": "2024-09-29@12:15:04"
+}

model.weights.h5

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

preprocessor.json

@@ -0,0 +1,27 @@
+{
+    "module": "keras_nlp.src.models.gemma.gemma_causal_lm_preprocessor",
+    "class_name": "GemmaCausalLMPreprocessor",
+    "config": {
+        "name": "gemma_causal_lm_preprocessor",
+        "trainable": true,
+        "dtype": "float32",
+        "tokenizer": {
+            "module": "keras_nlp.src.models.gemma.gemma_tokenizer",
+            "class_name": "GemmaTokenizer",
+            "config": {
+                "name": "gemma_tokenizer",
+                "trainable": true,
+                "dtype": "int32",
+                "proto": null,
+                "sequence_length": null,
+                "add_bos": false,
+                "add_eos": false
+            },
+            "registered_name": "keras_nlp>GemmaTokenizer"
+        },
+        "sequence_length": 512,
+        "add_start_token": true,
+        "add_end_token": true
+    },
+    "registered_name": "keras_nlp>GemmaCausalLMPreprocessor"
+}

task.json

@@ -0,0 +1,60 @@
+{
+    "module": "keras_nlp.src.models.gemma.gemma_causal_lm",
+    "class_name": "GemmaCausalLM",
+    "config": {
+        "backbone": {
+            "module": "keras_nlp.src.models.gemma.gemma_backbone",
+            "class_name": "GemmaBackbone",
+            "config": {
+                "name": "gemma_backbone",
+                "trainable": true,
+                "vocabulary_size": 256000,
+                "num_layers": 42,
+                "num_query_heads": 16,
+                "num_key_value_heads": 8,
+                "hidden_dim": 3584,
+                "intermediate_dim": 28672,
+                "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,
+                "attention_logit_soft_cap": 50,
+                "sliding_window_size": 4096,
+                "use_sliding_window_attention": true
+            },
+            "registered_name": "keras_nlp>GemmaBackbone"
+        },
+        "preprocessor": {
+            "module": "keras_nlp.src.models.gemma.gemma_causal_lm_preprocessor",
+            "class_name": "GemmaCausalLMPreprocessor",
+            "config": {
+                "name": "gemma_causal_lm_preprocessor",
+                "trainable": true,
+                "dtype": "float32",
+                "tokenizer": {
+                    "module": "keras_nlp.src.models.gemma.gemma_tokenizer",
+                    "class_name": "GemmaTokenizer",
+                    "config": {
+                        "name": "gemma_tokenizer",
+                        "trainable": true,
+                        "dtype": "int32",
+                        "proto": null,
+                        "sequence_length": null,
+                        "add_bos": false,
+                        "add_eos": false
+                    },
+                    "registered_name": "keras_nlp>GemmaTokenizer"
+                },
+                "sequence_length": 512,
+                "add_start_token": true,
+                "add_end_token": true
+            },
+            "registered_name": "keras_nlp>GemmaCausalLMPreprocessor"
+        },
+        "name": "gemma_causal_lm"
+    },
+    "registered_name": "keras_nlp>GemmaCausalLM"
+}

tokenizer.json

@@ -0,0 +1,14 @@
+{
+    "module": "keras_nlp.src.models.gemma.gemma_tokenizer",
+    "class_name": "GemmaTokenizer",
+    "config": {
+        "name": "gemma_tokenizer",
+        "trainable": true,
+        "dtype": "int32",
+        "proto": null,
+        "sequence_length": null,
+        "add_bos": false,
+        "add_eos": false
+    },
+    "registered_name": "keras_nlp>GemmaTokenizer"
+}