Skip to content
Merged
Show file tree
Hide file tree
Changes from 5 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 6 additions & 0 deletions scripts/docstrings.py
Original file line number Diff line number Diff line change
Expand Up @@ -217,10 +217,16 @@ 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("<") # <Foo object at 0x00000000>
and hasattr(value, "__class__") # it is an object
and hasattr(value, "get_config") # it is a Keras object
and not inspect.isclass(value) # ensure it's an instance, not a class
):
config = value.get_config()
init_args = [] # The __init__ arguments to render
Expand Down
204 changes: 204 additions & 0 deletions scripts/hub_master.py
Original file line number Diff line number Diff line change
Expand Up @@ -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",
Expand Down Expand Up @@ -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",
Expand Down Expand Up @@ -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",
Expand Down Expand Up @@ -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",
Expand Down Expand Up @@ -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",
Expand Down