diff --git a/scripts/autogen.py b/scripts/autogen.py index af42f87dae..c4b31ea09e 100644 --- a/scripts/autogen.py +++ b/scripts/autogen.py @@ -47,7 +47,7 @@ "keras": f"{KERAS_TEAM_GH}/keras/tree/v3.3.2/", "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 index 0571fb1baf..75e9eec03b 100644 --- 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", ], }, {