Skip to content
Merged
Show file tree
Hide file tree
Changes from all 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
5 changes: 5 additions & 0 deletions scripts/docstrings.py
Original file line number Diff line number Diff line change
Expand Up @@ -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("<") # <Foo object at 0x00000000>
and hasattr(value, "__class__") # it is an object
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