Add T5Gemma2 model to Hub

keras_hub/api/models/__init__.py

@@ -762,6 +762,18 @@
 from keras_hub.src.models.t5gemma.t5gemma_tokenizer import (
     T5GemmaTokenizer as T5GemmaTokenizer,
 )
+from keras_hub.src.models.t5gemma2.t5gemma2_backbone import (
+    T5Gemma2Backbone as T5Gemma2Backbone,
+)
+from keras_hub.src.models.t5gemma2.t5gemma2_seq_2_seq_lm import (
+    T5Gemma2Seq2SeqLM as T5Gemma2Seq2SeqLM,
+)
+from keras_hub.src.models.t5gemma2.t5gemma2_seq_2_seq_lm_preprocessor import (
+    T5Gemma2Seq2SeqLMPreprocessor as T5Gemma2Seq2SeqLMPreprocessor,
+)
+from keras_hub.src.models.t5gemma2.t5gemma2_tokenizer import (
+    T5Gemma2Tokenizer as T5Gemma2Tokenizer,
+)
 from keras_hub.src.models.task import Task as Task
 from keras_hub.src.models.text_classifier import TextClassifier as Classifier
 from keras_hub.src.models.text_classifier import (

keras_hub/api/tokenizers/__init__.py

@@ -115,6 +115,9 @@
 from keras_hub.src.models.t5gemma.t5gemma_tokenizer import (
     T5GemmaTokenizer as T5GemmaTokenizer,
 )
+from keras_hub.src.models.t5gemma2.t5gemma2_tokenizer import (
+    T5Gemma2Tokenizer as T5Gemma2Tokenizer,
+)
 from keras_hub.src.models.video_prism.video_prism_tokenizer import (
     VideoPrismTokenizer as VideoPrismTokenizer,
 )

keras_hub/src/models/t5gemma2/__init__.py

@@ -0,0 +1,5 @@
+from keras_hub.src.models.t5gemma2.t5gemma2_backbone import T5Gemma2Backbone
+from keras_hub.src.models.t5gemma2.t5gemma2_presets import backbone_presets
+from keras_hub.src.utils.preset_utils import register_presets
+
+register_presets(backbone_presets, T5Gemma2Backbone)