Skip to content

Commit f890f49

Browse files
sachinprasadhssachinprasadhs
authored
Add Depth Anything,MobilenetV5,ParSeq, Qwen3Moe, SmollLM3 Hub model API details (#2204)
* add new Hub model API details * address gemini review * nit * nit * make changes to docstring generation * address comments
1 parent b6949c5 commit f890f49

File tree

2 files changed

+209
-0
lines changed

2 files changed

+209
-0
lines changed

‎scripts/docstrings.py‎

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -217,6 +217,11 @@ def __repr__(self):
217217
# Render the function name instead
218218
return ReprWrapper(value.__name__)
219219

220+
if inspect.isclass(value):
221+
# Render classes as module.ClassName to produce a valid python
222+
# dotted-name expression in the fake signature (black can parse it).
223+
return ReprWrapper(value.__module__ + "." + value.__name__)
224+
220225
if (
221226
repr(value).startswith("<") # <Foo object at 0x00000000>
222227
and hasattr(value, "__class__") # it is an object

‎scripts/hub_master.py‎

Lines changed: 204 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -781,6 +781,38 @@
781781
},
782782
],
783783
},
784+
{
785+
"path": "depth_anything/",
786+
"title": "DepthAnything",
787+
"toc": True,
788+
"children": [
789+
{
790+
"path": "depth_anything_image_converter",
791+
"title": "DepthAnythingImageConverter",
792+
"generate": [
793+
"keras_hub.layers.DepthAnythingImageConverter",
794+
"keras_hub.layers.DepthAnythingImageConverter.from_preset",
795+
],
796+
},
797+
{
798+
"path": "depth_anything_backbone",
799+
"title": "DepthAnythingBackbone model",
800+
"generate": [
801+
"keras_hub.models.DepthAnythingBackbone",
802+
"keras_hub.models.DepthAnythingBackbone.from_preset",
803+
],
804+
},
805+
{
806+
"path": "depth_anything_depth_estimator",
807+
"title": "DepthAnythingDepthEstimator model",
808+
"generate": [
809+
"keras_hub.models.DepthAnythingDepthEstimator",
810+
"keras_hub.models.DepthAnythingDepthEstimator.from_preset",
811+
"keras_hub.models.DepthAnythingDepthEstimator.backbone",
812+
],
813+
},
814+
],
815+
},
784816
{
785817
"path": "dinov2/",
786818
"title": "DINOV2",
@@ -1633,6 +1665,46 @@
16331665
},
16341666
],
16351667
},
1668+
{
1669+
"path": "mobilenetv5/",
1670+
"title": "MobileNetV5",
1671+
"toc": True,
1672+
"children": [
1673+
{
1674+
"path": "mobilenetv5_image_converter",
1675+
"title": "MobileNetV5ImageConverter",
1676+
"generate": [
1677+
"keras_hub.layers.MobileNetV5ImageConverter",
1678+
"keras_hub.layers.MobileNetV5ImageConverter.from_preset",
1679+
],
1680+
},
1681+
{
1682+
"path": "mobilenetv5_backbone",
1683+
"title": "MobileNetV5Backbone model",
1684+
"generate": [
1685+
"keras_hub.models.MobileNetV5Backbone",
1686+
"keras_hub.models.MobileNetV5Backbone.from_preset",
1687+
],
1688+
},
1689+
{
1690+
"path": "mobilenetv5_image_classifier",
1691+
"title": "MobileNetV5ImageClassifier model",
1692+
"generate": [
1693+
"keras_hub.models.MobileNetV5ImageClassifier",
1694+
"keras_hub.models.MobileNetV5ImageClassifier.backbone",
1695+
"keras_hub.models.MobileNetV5ImageClassifier.preprocessor",
1696+
],
1697+
},
1698+
{
1699+
"path": "mobilenetv5_image_classifier_preprocessor",
1700+
"title": "MobileNetV5ImageClassifierPreprocessor layer",
1701+
"generate": [
1702+
"keras_hub.models.MobileNetV5ImageClassifierPreprocessor",
1703+
"keras_hub.models.MobileNetV5ImageClassifierPreprocessor.image_converter",
1704+
],
1705+
},
1706+
],
1707+
},
16361708
{
16371709
"path": "moonshine/",
16381710
"title": "Moonshine",
@@ -1773,6 +1845,49 @@
17731845
},
17741846
],
17751847
},
1848+
{
1849+
"path": "parseq/",
1850+
"title": "PARSeq",
1851+
"toc": True,
1852+
"children": [
1853+
{
1854+
"path": "parseq_tokenizer",
1855+
"title": "PARSeqTokenizer",
1856+
"generate": [
1857+
"keras_hub.tokenizers.PARSeqTokenizer",
1858+
"keras_hub.tokenizers.PARSeqTokenizer.from_preset",
1859+
],
1860+
},
1861+
{
1862+
"path": "parseq_backbone",
1863+
"title": "PARSeqBackbone model",
1864+
"generate": [
1865+
"keras_hub.models.PARSeqBackbone",
1866+
"keras_hub.models.PARSeqBackbone.from_preset",
1867+
],
1868+
},
1869+
{
1870+
"path": "parseq_causal_lm",
1871+
"title": "PARSeqCausalLM model",
1872+
"generate": [
1873+
"keras_hub.models.PARSeqCausalLM",
1874+
"keras_hub.models.PARSeqCausalLM.from_preset",
1875+
"keras_hub.models.PARSeqCausalLM.generate",
1876+
"keras_hub.models.PARSeqCausalLM.backbone",
1877+
"keras_hub.models.PARSeqCausalLM.preprocessor",
1878+
],
1879+
},
1880+
{
1881+
"path": "parseq_causal_lmpreprocessor",
1882+
"title": "PARSeqCausalLMPreprocessor layer",
1883+
"generate": [
1884+
"keras_hub.models.PARSeqCausalLMPreprocessor",
1885+
"keras_hub.models.PARSeqCausalLMPreprocessor.from_preset",
1886+
"keras_hub.models.PARSeqCausalLMPreprocessor.tokenizer",
1887+
],
1888+
},
1889+
],
1890+
},
17761891
{
17771892
"path": "phi3/",
17781893
"title": "Phi3",
@@ -1907,6 +2022,51 @@
19072022
},
19082023
],
19092024
},
2025+
{
2026+
"path": "qwen3_moe/",
2027+
"title": "Qwen3Moe",
2028+
"toc": True,
2029+
"children": [
2030+
{
2031+
"path": "qwen3_moe_tokenizer",
2032+
"title": "Qwen3MoeTokenizer",
2033+
"generate": [
2034+
"keras_hub.tokenizers.Qwen3MoeTokenizer",
2035+
"keras_hub.tokenizers.Qwen3MoeTokenizer.from_preset",
2036+
],
2037+
},
2038+
{
2039+
"path": "qwen3_moe_backbone",
2040+
"title": "Qwen3MoeBackbone model",
2041+
"generate": [
2042+
"keras_hub.models.Qwen3MoeBackbone",
2043+
"keras_hub.models.Qwen3MoeBackbone.from_preset",
2044+
"keras_hub.models.Qwen3MoeBackbone.token_embedding",
2045+
"keras_hub.models.Qwen3MoeBackbone.enable_lora",
2046+
],
2047+
},
2048+
{
2049+
"path": "qwen3_moe_causal_lm",
2050+
"title": "Qwen3MoeCausalLM model",
2051+
"generate": [
2052+
"keras_hub.models.Qwen3MoeCausalLM",
2053+
"keras_hub.models.Qwen3MoeCausalLM.from_preset",
2054+
"keras_hub.models.Qwen3MoeCausalLM.generate",
2055+
"keras_hub.models.Qwen3MoeCausalLM.backbone",
2056+
"keras_hub.models.Qwen3MoeCausalLM.preprocessor",
2057+
],
2058+
},
2059+
{
2060+
"path": "qwen3_moe_causal_lm_preprocessor",
2061+
"title": "Qwen3MoeCausalLMPreprocessor layer",
2062+
"generate": [
2063+
"keras_hub.models.Qwen3MoeCausalLMPreprocessor",
2064+
"keras_hub.models.Qwen3MoeCausalLMPreprocessor.from_preset",
2065+
"keras_hub.models.Qwen3MoeCausalLMPreprocessor.tokenizer",
2066+
],
2067+
},
2068+
],
2069+
},
19102070
{
19112071
"path": "qwen_moe/",
19122072
"title": "QwenMoe",
@@ -2236,6 +2396,50 @@
22362396
},
22372397
],
22382398
},
2399+
{
2400+
"path": "smollm3/",
2401+
"title": "SmolLM3",
2402+
"toc": True,
2403+
"children": [
2404+
{
2405+
"path": "smollm3_tokenizer",
2406+
"title": "SmolLM3Tokenizer",
2407+
"generate": [
2408+
"keras_hub.models.SmolLM3Tokenizer",
2409+
"keras_hub.models.SmolLM3Tokenizer.from_preset",
2410+
],
2411+
},
2412+
{
2413+
"path": "smollm3_backbone",
2414+
"title": "SmolLM3Backbone model",
2415+
"generate": [
2416+
"keras_hub.models.SmolLM3Backbone",
2417+
"keras_hub.models.SmolLM3Backbone.from_preset",
2418+
"keras_hub.models.SmolLM3Backbone.token_embedding",
2419+
],
2420+
},
2421+
{
2422+
"path": "smollm3_causal_lm",
2423+
"title": "SmolLM3CausalLM model",
2424+
"generate": [
2425+
"keras_hub.models.SmolLM3CausalLM",
2426+
"keras_hub.models.SmolLM3CausalLM.from_preset",
2427+
"keras_hub.models.SmolLM3CausalLM.generate",
2428+
"keras_hub.models.SmolLM3CausalLM.backbone",
2429+
"keras_hub.models.SmolLM3CausalLM.preprocessor",
2430+
],
2431+
},
2432+
{
2433+
"path": "smollm3_causal_lm_preprocessor",
2434+
"title": "SmolLM3CausalLMPreprocessor layer",
2435+
"generate": [
2436+
"keras_hub.models.SmolLM3CausalLMPreprocessor",
2437+
"keras_hub.models.SmolLM3CausalLMPreprocessor.from_preset",
2438+
"keras_hub.models.SmolLM3CausalLMPreprocessor.tokenizer",
2439+
],
2440+
},
2441+
],
2442+
},
22392443
{
22402444
"path": "stable_diffusion_3/",
22412445
"title": "Stable Diffusion 3",

0 commit comments

Comments
 (0)