Update keras.io for KerasNLP 0.10.0 (#1846)

keras-team · Apr 29, 2024 · ce18bbb · ce18bbb
1 parent 80af9a7
commit ce18bbb
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/scripts/autogen.py b/scripts/autogen.py
@@ -47,7 +47,7 @@
     "keras": f"{KERAS_TEAM_GH}/keras/tree/v3.3.3/",
     "keras_tuner": f"{KERAS_TEAM_GH}/keras-tuner/tree/v1.4.7/",
     "keras_cv": f"{KERAS_TEAM_GH}/keras-cv/tree/v0.8.2/",
-    "keras_nlp": f"{KERAS_TEAM_GH}/keras-nlp/tree/v0.9.3/",
+    "keras_nlp": f"{KERAS_TEAM_GH}/keras-nlp/tree/v0.10.0/",
     "tf_keras": f"{KERAS_TEAM_GH}/tf-keras/tree/v2.16.0/",
 }
 USE_MULTIPROCESSING = False

diff --git a/scripts/nlp_api_master.py b/scripts/nlp_api_master.py
@@ -24,6 +24,7 @@
                 "keras_nlp.models.CausalLM.from_preset",
                 "keras_nlp.models.CausalLM.compile",
                 "keras_nlp.models.CausalLM.generate",
+                "keras_nlp.models.CausalLM.save_to_preset",
                 "keras_nlp.models.CausalLM.preprocessor",
                 "keras_nlp.models.CausalLM.backbone",
             ],
@@ -36,6 +37,7 @@
                 "keras_nlp.models.Seq2SeqLM.from_preset",
                 "keras_nlp.models.Seq2SeqLM.compile",
                 "keras_nlp.models.Seq2SeqLM.generate",
+                "keras_nlp.models.Seq2SeqLM.save_to_preset",
                 "keras_nlp.models.Seq2SeqLM.preprocessor",
                 "keras_nlp.models.Seq2SeqLM.backbone",
             ],
@@ -47,6 +49,7 @@
                 "keras_nlp.models.Classifier",
                 "keras_nlp.models.Classifier.from_preset",
                 "keras_nlp.models.Classifier.compile",
+                "keras_nlp.models.Classifier.save_to_preset",
                 "keras_nlp.models.Classifier.preprocessor",
                 "keras_nlp.models.Classifier.backbone",
             ],
@@ -58,6 +61,7 @@
                 "keras_nlp.models.MaskedLM",
                 "keras_nlp.models.MaskedLM.from_preset",
                 "keras_nlp.models.MaskedLM.compile",
+                "keras_nlp.models.MaskedLM.save_to_preset",
                 "keras_nlp.models.MaskedLM.preprocessor",
                 "keras_nlp.models.MaskedLM.backbone",
             ],
@@ -68,6 +72,7 @@
             "generate": [
                 "keras_nlp.models.Task",
                 "keras_nlp.models.Task.from_preset",
+                "keras_nlp.models.Task.save_to_preset",
                 "keras_nlp.models.Task.preprocessor",
                 "keras_nlp.models.Task.backbone",
             ],
@@ -78,6 +83,7 @@
             "generate": [
                 "keras_nlp.models.Preprocessor",
                 "keras_nlp.models.Preprocessor.from_preset",
+                "keras_nlp.models.Preprocessor.save_to_preset",
                 "keras_nlp.models.Preprocessor.tokenizer",
             ],
         },
@@ -87,6 +93,7 @@
             "generate": [
                 "keras_nlp.models.Tokenizer",
                 "keras_nlp.models.Tokenizer.from_preset",
+                "keras_nlp.models.Tokenizer.save_to_preset",
             ],
         },
         {