diff --git a/scripts/docstrings.py b/scripts/docstrings.py index 3078e4a7b7..169b46d1cb 100644 --- a/scripts/docstrings.py +++ b/scripts/docstrings.py @@ -217,6 +217,11 @@ def __repr__(self): # Render the function name instead return ReprWrapper(value.__name__) + if inspect.isclass(value): + # Render classes as module.ClassName to produce a valid python + # dotted-name expression in the fake signature (black can parse it). + return ReprWrapper(value.__module__ + "." + value.__name__) + if ( repr(value).startswith("<") # and hasattr(value, "__class__") # it is an object diff --git a/scripts/hub_master.py b/scripts/hub_master.py index fc323ebe03..987282a048 100644 --- a/scripts/hub_master.py +++ b/scripts/hub_master.py @@ -739,6 +739,38 @@ }, ], }, + { + "path": "depth_anything/", + "title": "DepthAnything", + "toc": True, + "children": [ + { + "path": "depth_anything_image_converter", + "title": "DepthAnythingImageConverter", + "generate": [ + "keras_hub.layers.DepthAnythingImageConverter", + "keras_hub.layers.DepthAnythingImageConverter.from_preset", + ], + }, + { + "path": "depth_anything_backbone", + "title": "DepthAnythingBackbone model", + "generate": [ + "keras_hub.models.DepthAnythingBackbone", + "keras_hub.models.DepthAnythingBackbone.from_preset", + ], + }, + { + "path": "depth_anything_depth_estimator", + "title": "DepthAnythingDepthEstimator model", + "generate": [ + "keras_hub.models.DepthAnythingDepthEstimator", + "keras_hub.models.DepthAnythingDepthEstimator.from_preset", + "keras_hub.models.DepthAnythingDepthEstimator.backbone", + ], + }, + ], + }, { "path": "dinov2/", "title": "DINOV2", @@ -1591,6 +1623,46 @@ }, ], }, + { + "path": "mobilenetv5/", + "title": "MobileNetV5", + "toc": True, + "children": [ + { + "path": "mobilenetv5_image_converter", + "title": "MobileNetV5ImageConverter", + "generate": [ + "keras_hub.layers.MobileNetV5ImageConverter", + "keras_hub.layers.MobileNetV5ImageConverter.from_preset", + ], + }, + { + "path": "mobilenetv5_backbone", + "title": "MobileNetV5Backbone model", + "generate": [ + "keras_hub.models.MobileNetV5Backbone", + "keras_hub.models.MobileNetV5Backbone.from_preset", + ], + }, + { + "path": "mobilenetv5_image_classifier", + "title": "MobileNetV5ImageClassifier model", + "generate": [ + "keras_hub.models.MobileNetV5ImageClassifier", + "keras_hub.models.MobileNetV5ImageClassifier.backbone", + "keras_hub.models.MobileNetV5ImageClassifier.preprocessor", + ], + }, + { + "path": "mobilenetv5_image_classifier_preprocessor", + "title": "MobileNetV5ImageClassifierPreprocessor layer", + "generate": [ + "keras_hub.models.MobileNetV5ImageClassifierPreprocessor", + "keras_hub.models.MobileNetV5ImageClassifierPreprocessor.image_converter", + ], + }, + ], + }, { "path": "moonshine/", "title": "Moonshine", @@ -1731,6 +1803,49 @@ }, ], }, + { + "path": "parseq/", + "title": "PARSeq", + "toc": True, + "children": [ + { + "path": "parseq_tokenizer", + "title": "PARSeqTokenizer", + "generate": [ + "keras_hub.tokenizers.PARSeqTokenizer", + "keras_hub.tokenizers.PARSeqTokenizer.from_preset", + ], + }, + { + "path": "parseq_backbone", + "title": "PARSeqBackbone model", + "generate": [ + "keras_hub.models.PARSeqBackbone", + "keras_hub.models.PARSeqBackbone.from_preset", + ], + }, + { + "path": "parseq_causal_lm", + "title": "PARSeqCausalLM model", + "generate": [ + "keras_hub.models.PARSeqCausalLM", + "keras_hub.models.PARSeqCausalLM.from_preset", + "keras_hub.models.PARSeqCausalLM.generate", + "keras_hub.models.PARSeqCausalLM.backbone", + "keras_hub.models.PARSeqCausalLM.preprocessor", + ], + }, + { + "path": "parseq_causal_lmpreprocessor", + "title": "PARSeqCausalLMPreprocessor layer", + "generate": [ + "keras_hub.models.PARSeqCausalLMPreprocessor", + "keras_hub.models.PARSeqCausalLMPreprocessor.from_preset", + "keras_hub.models.PARSeqCausalLMPreprocessor.tokenizer", + ], + }, + ], + }, { "path": "phi3/", "title": "Phi3", @@ -1865,6 +1980,51 @@ }, ], }, + { + "path": "qwen3_moe/", + "title": "Qwen3Moe", + "toc": True, + "children": [ + { + "path": "qwen3_moe_tokenizer", + "title": "Qwen3MoeTokenizer", + "generate": [ + "keras_hub.tokenizers.Qwen3MoeTokenizer", + "keras_hub.tokenizers.Qwen3MoeTokenizer.from_preset", + ], + }, + { + "path": "qwen3_moe_backbone", + "title": "Qwen3MoeBackbone model", + "generate": [ + "keras_hub.models.Qwen3MoeBackbone", + "keras_hub.models.Qwen3MoeBackbone.from_preset", + "keras_hub.models.Qwen3MoeBackbone.token_embedding", + "keras_hub.models.Qwen3MoeBackbone.enable_lora", + ], + }, + { + "path": "qwen3_moe_causal_lm", + "title": "Qwen3MoeCausalLM model", + "generate": [ + "keras_hub.models.Qwen3MoeCausalLM", + "keras_hub.models.Qwen3MoeCausalLM.from_preset", + "keras_hub.models.Qwen3MoeCausalLM.generate", + "keras_hub.models.Qwen3MoeCausalLM.backbone", + "keras_hub.models.Qwen3MoeCausalLM.preprocessor", + ], + }, + { + "path": "qwen3_moe_causal_lm_preprocessor", + "title": "Qwen3MoeCausalLMPreprocessor layer", + "generate": [ + "keras_hub.models.Qwen3MoeCausalLMPreprocessor", + "keras_hub.models.Qwen3MoeCausalLMPreprocessor.from_preset", + "keras_hub.models.Qwen3MoeCausalLMPreprocessor.tokenizer", + ], + }, + ], + }, { "path": "qwen_moe/", "title": "QwenMoe", @@ -2194,6 +2354,50 @@ }, ], }, + { + "path": "smollm3/", + "title": "SmolLM3", + "toc": True, + "children": [ + { + "path": "smollm3_tokenizer", + "title": "SmolLM3Tokenizer", + "generate": [ + "keras_hub.models.SmolLM3Tokenizer", + "keras_hub.models.SmolLM3Tokenizer.from_preset", + ], + }, + { + "path": "smollm3_backbone", + "title": "SmolLM3Backbone model", + "generate": [ + "keras_hub.models.SmolLM3Backbone", + "keras_hub.models.SmolLM3Backbone.from_preset", + "keras_hub.models.SmolLM3Backbone.token_embedding", + ], + }, + { + "path": "smollm3_causal_lm", + "title": "SmolLM3CausalLM model", + "generate": [ + "keras_hub.models.SmolLM3CausalLM", + "keras_hub.models.SmolLM3CausalLM.from_preset", + "keras_hub.models.SmolLM3CausalLM.generate", + "keras_hub.models.SmolLM3CausalLM.backbone", + "keras_hub.models.SmolLM3CausalLM.preprocessor", + ], + }, + { + "path": "smollm3_causal_lm_preprocessor", + "title": "SmolLM3CausalLMPreprocessor layer", + "generate": [ + "keras_hub.models.SmolLM3CausalLMPreprocessor", + "keras_hub.models.SmolLM3CausalLMPreprocessor.from_preset", + "keras_hub.models.SmolLM3CausalLMPreprocessor.tokenizer", + ], + }, + ], + }, { "path": "stable_diffusion_3/", "title": "Stable Diffusion 3",