keras-hub 0.20.0.dev1__py3-none-any.whl → 0.21.0.dev1__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (105) hide show
  1. keras_hub/__init__.py +15 -33
  2. keras_hub/layers/__init__.py +134 -0
  3. keras_hub/metrics/__init__.py +11 -0
  4. keras_hub/models/__init__.py +642 -0
  5. keras_hub/samplers/__init__.py +18 -0
  6. keras_hub/src/layers/modeling/reversible_embedding.py +25 -35
  7. keras_hub/src/layers/preprocessing/image_converter.py +1 -0
  8. keras_hub/src/layers/preprocessing/random_deletion.py +1 -1
  9. keras_hub/src/layers/preprocessing/random_swap.py +1 -1
  10. keras_hub/src/models/audio_to_text.py +66 -0
  11. keras_hub/src/models/audio_to_text_preprocessor.py +80 -0
  12. keras_hub/src/models/backbone.py +5 -2
  13. keras_hub/src/models/cspnet/cspnet_backbone.py +51 -26
  14. keras_hub/src/models/cspnet/cspnet_presets.py +38 -3
  15. keras_hub/src/models/falcon/falcon_backbone.py +1 -1
  16. keras_hub/src/models/gemma/gemma_presets.py +10 -10
  17. keras_hub/src/models/gemma3/gemma3_causal_lm_preprocessor.py +3 -2
  18. keras_hub/src/models/gemma3/gemma3_presets.py +8 -8
  19. keras_hub/src/models/gemma3/gemma3_vision_encoder.py +1 -1
  20. keras_hub/src/models/llama/llama_attention.py +24 -6
  21. keras_hub/src/models/llama/llama_backbone.py +50 -16
  22. keras_hub/src/models/llama/llama_decoder.py +20 -3
  23. keras_hub/src/models/llama/llama_presets.py +3 -3
  24. keras_hub/src/models/llama/llama_rotary_embedding.py +180 -0
  25. keras_hub/src/models/llama3/llama3_backbone.py +10 -2
  26. keras_hub/src/models/llama3/llama3_presets.py +84 -2
  27. keras_hub/src/models/mistral/mistral_presets.py +3 -3
  28. keras_hub/src/models/mixtral/__init__.py +5 -0
  29. keras_hub/src/models/mixtral/mixtral_attention.py +252 -0
  30. keras_hub/src/models/mixtral/mixtral_backbone.py +207 -0
  31. keras_hub/src/models/mixtral/mixtral_causal_lm.py +281 -0
  32. keras_hub/src/models/mixtral/mixtral_causal_lm_preprocessor.py +76 -0
  33. keras_hub/src/models/mixtral/mixtral_decoder.py +494 -0
  34. keras_hub/src/models/mixtral/mixtral_layer_norm.py +34 -0
  35. keras_hub/src/models/mixtral/mixtral_presets.py +26 -0
  36. keras_hub/src/models/mixtral/mixtral_tokenizer.py +21 -0
  37. keras_hub/src/models/moonshine/__init__.py +5 -0
  38. keras_hub/src/models/moonshine/moonshine_audio_converter.py +301 -0
  39. keras_hub/src/models/moonshine/moonshine_audio_to_text.py +383 -0
  40. keras_hub/src/models/moonshine/moonshine_audio_to_text_preprocessor.py +272 -0
  41. keras_hub/src/models/moonshine/moonshine_backbone.py +478 -0
  42. keras_hub/src/models/moonshine/moonshine_decoder.py +313 -0
  43. keras_hub/src/models/moonshine/moonshine_encoder.py +212 -0
  44. keras_hub/src/models/moonshine/moonshine_layers.py +239 -0
  45. keras_hub/src/models/moonshine/moonshine_multi_head_attention.py +355 -0
  46. keras_hub/src/models/moonshine/moonshine_presets.py +25 -0
  47. keras_hub/src/models/moonshine/moonshine_tokenizer.py +62 -0
  48. keras_hub/src/models/pali_gemma/pali_gemma_presets.py +11 -11
  49. keras_hub/src/models/pali_gemma/pali_gemma_vit.py +1 -1
  50. keras_hub/src/models/qwen/__init__.py +4 -0
  51. keras_hub/src/models/qwen/qwen_attention.py +3 -1
  52. keras_hub/src/models/qwen/qwen_backbone.py +8 -1
  53. keras_hub/src/models/qwen/qwen_causal_lm.py +7 -0
  54. keras_hub/src/models/qwen/qwen_causal_lm_preprocessor.py +7 -0
  55. keras_hub/src/models/qwen/qwen_presets.py +61 -0
  56. keras_hub/src/models/qwen/qwen_tokenizer.py +9 -0
  57. keras_hub/src/models/qwen_moe/__init__.py +5 -0
  58. keras_hub/src/models/qwen_moe/qwen_moe_attention.py +375 -0
  59. keras_hub/src/models/qwen_moe/qwen_moe_backbone.py +373 -0
  60. keras_hub/src/models/qwen_moe/qwen_moe_causal_lm.py +350 -0
  61. keras_hub/src/models/qwen_moe/qwen_moe_causal_lm_preprocessor.py +17 -0
  62. keras_hub/src/models/qwen_moe/qwen_moe_decoder.py +625 -0
  63. keras_hub/src/models/qwen_moe/qwen_moe_layernorm.py +32 -0
  64. keras_hub/src/models/qwen_moe/qwen_moe_presets.py +15 -0
  65. keras_hub/src/models/qwen_moe/qwen_moe_tokenizer.py +46 -0
  66. keras_hub/src/models/retinanet/retinanet_image_converter.py +0 -13
  67. keras_hub/src/models/retinanet/retinanet_presets.py +2 -2
  68. keras_hub/src/models/segformer/segformer_image_segmenter_preprocessor.py +0 -18
  69. keras_hub/src/models/segformer/segformer_presets.py +12 -12
  70. keras_hub/src/models/stable_diffusion_3/stable_diffusion_3_backbone.py +6 -0
  71. keras_hub/src/models/task.py +5 -2
  72. keras_hub/src/models/xception/__init__.py +5 -0
  73. keras_hub/src/models/xception/xception_backbone.py +188 -0
  74. keras_hub/src/models/xception/xception_image_classifier.py +12 -0
  75. keras_hub/src/models/xception/xception_image_classifier_preprocessor.py +14 -0
  76. keras_hub/src/models/xception/xception_image_converter.py +8 -0
  77. keras_hub/src/models/xception/xception_presets.py +14 -0
  78. keras_hub/src/tests/mocks/mock_gemma3_tokenizer.py +155 -0
  79. keras_hub/src/utils/coco/__init__.py +0 -0
  80. keras_hub/src/utils/coco/coco_utils.py +133 -0
  81. keras_hub/src/utils/imagenet/imagenet_utils.py +36 -0
  82. keras_hub/src/utils/keras_utils.py +11 -0
  83. keras_hub/src/utils/preset_utils.py +70 -10
  84. keras_hub/src/utils/tensor_utils.py +27 -1
  85. keras_hub/src/utils/timm/convert_cspnet.py +94 -23
  86. keras_hub/src/utils/timm/preset_loader.py +6 -6
  87. keras_hub/src/utils/transformers/convert_llama3.py +21 -1
  88. keras_hub/src/utils/transformers/convert_mixtral.py +139 -0
  89. keras_hub/src/utils/transformers/convert_qwen.py +1 -0
  90. keras_hub/src/utils/transformers/convert_qwen_moe.py +253 -0
  91. keras_hub/src/utils/transformers/preset_loader.py +6 -0
  92. keras_hub/src/{version_utils.py → version.py} +1 -1
  93. keras_hub/tokenizers/__init__.py +117 -0
  94. keras_hub/utils/__init__.py +21 -0
  95. {keras_hub-0.20.0.dev1.dist-info → keras_hub-0.21.0.dev1.dist-info}/METADATA +6 -20
  96. {keras_hub-0.20.0.dev1.dist-info → keras_hub-0.21.0.dev1.dist-info}/RECORD +98 -55
  97. {keras_hub-0.20.0.dev1.dist-info → keras_hub-0.21.0.dev1.dist-info}/WHEEL +1 -1
  98. keras_hub/api/__init__.py +0 -15
  99. keras_hub/api/layers/__init__.py +0 -86
  100. keras_hub/api/metrics/__init__.py +0 -11
  101. keras_hub/api/models/__init__.py +0 -416
  102. keras_hub/api/samplers/__init__.py +0 -16
  103. keras_hub/api/tokenizers/__init__.py +0 -58
  104. keras_hub/api/utils/__init__.py +0 -9
  105. {keras_hub-0.20.0.dev1.dist-info → keras_hub-0.21.0.dev1.dist-info}/top_level.txt +0 -0
@@ -1,416 +0,0 @@
1
- """DO NOT EDIT.
2
-
3
- This file was autogenerated. Do not edit it by hand,
4
- since your modifications would be overwritten.
5
- """
6
-
7
- from keras_hub.src.models.albert.albert_backbone import AlbertBackbone
8
- from keras_hub.src.models.albert.albert_masked_lm import AlbertMaskedLM
9
- from keras_hub.src.models.albert.albert_masked_lm_preprocessor import (
10
- AlbertMaskedLMPreprocessor,
11
- )
12
- from keras_hub.src.models.albert.albert_text_classifier import (
13
- AlbertTextClassifier,
14
- )
15
- from keras_hub.src.models.albert.albert_text_classifier import (
16
- AlbertTextClassifier as AlbertClassifier,
17
- )
18
- from keras_hub.src.models.albert.albert_text_classifier_preprocessor import (
19
- AlbertTextClassifierPreprocessor,
20
- )
21
- from keras_hub.src.models.albert.albert_text_classifier_preprocessor import (
22
- AlbertTextClassifierPreprocessor as AlbertPreprocessor,
23
- )
24
- from keras_hub.src.models.albert.albert_tokenizer import AlbertTokenizer
25
- from keras_hub.src.models.backbone import Backbone
26
- from keras_hub.src.models.bart.bart_backbone import BartBackbone
27
- from keras_hub.src.models.bart.bart_seq_2_seq_lm import BartSeq2SeqLM
28
- from keras_hub.src.models.bart.bart_seq_2_seq_lm_preprocessor import (
29
- BartSeq2SeqLMPreprocessor,
30
- )
31
- from keras_hub.src.models.bart.bart_tokenizer import BartTokenizer
32
- from keras_hub.src.models.basnet.basnet import BASNetImageSegmenter
33
- from keras_hub.src.models.basnet.basnet_backbone import BASNetBackbone
34
- from keras_hub.src.models.basnet.basnet_preprocessor import BASNetPreprocessor
35
- from keras_hub.src.models.bert.bert_backbone import BertBackbone
36
- from keras_hub.src.models.bert.bert_masked_lm import BertMaskedLM
37
- from keras_hub.src.models.bert.bert_masked_lm_preprocessor import (
38
- BertMaskedLMPreprocessor,
39
- )
40
- from keras_hub.src.models.bert.bert_text_classifier import BertTextClassifier
41
- from keras_hub.src.models.bert.bert_text_classifier import (
42
- BertTextClassifier as BertClassifier,
43
- )
44
- from keras_hub.src.models.bert.bert_text_classifier_preprocessor import (
45
- BertTextClassifierPreprocessor,
46
- )
47
- from keras_hub.src.models.bert.bert_text_classifier_preprocessor import (
48
- BertTextClassifierPreprocessor as BertPreprocessor,
49
- )
50
- from keras_hub.src.models.bert.bert_tokenizer import BertTokenizer
51
- from keras_hub.src.models.bloom.bloom_backbone import BloomBackbone
52
- from keras_hub.src.models.bloom.bloom_causal_lm import BloomCausalLM
53
- from keras_hub.src.models.bloom.bloom_causal_lm_preprocessor import (
54
- BloomCausalLMPreprocessor,
55
- )
56
- from keras_hub.src.models.bloom.bloom_tokenizer import BloomTokenizer
57
- from keras_hub.src.models.causal_lm import CausalLM
58
- from keras_hub.src.models.causal_lm_preprocessor import CausalLMPreprocessor
59
- from keras_hub.src.models.clip.clip_backbone import CLIPBackbone
60
- from keras_hub.src.models.clip.clip_preprocessor import CLIPPreprocessor
61
- from keras_hub.src.models.clip.clip_text_encoder import CLIPTextEncoder
62
- from keras_hub.src.models.clip.clip_tokenizer import CLIPTokenizer
63
- from keras_hub.src.models.clip.clip_vision_encoder import CLIPVisionEncoder
64
- from keras_hub.src.models.cspnet.cspnet_backbone import CSPNetBackbone
65
- from keras_hub.src.models.cspnet.cspnet_image_classifier import (
66
- CSPNetImageClassifier,
67
- )
68
- from keras_hub.src.models.cspnet.cspnet_image_classifier_preprocessor import (
69
- CSPNetImageClassifierPreprocessor,
70
- )
71
- from keras_hub.src.models.deberta_v3.deberta_v3_backbone import (
72
- DebertaV3Backbone,
73
- )
74
- from keras_hub.src.models.deberta_v3.deberta_v3_masked_lm import (
75
- DebertaV3MaskedLM,
76
- )
77
- from keras_hub.src.models.deberta_v3.deberta_v3_masked_lm_preprocessor import (
78
- DebertaV3MaskedLMPreprocessor,
79
- )
80
- from keras_hub.src.models.deberta_v3.deberta_v3_text_classifier import (
81
- DebertaV3TextClassifier,
82
- )
83
- from keras_hub.src.models.deberta_v3.deberta_v3_text_classifier import (
84
- DebertaV3TextClassifier as DebertaV3Classifier,
85
- )
86
- from keras_hub.src.models.deberta_v3.deberta_v3_text_classifier_preprocessor import (
87
- DebertaV3TextClassifierPreprocessor,
88
- )
89
- from keras_hub.src.models.deberta_v3.deberta_v3_text_classifier_preprocessor import (
90
- DebertaV3TextClassifierPreprocessor as DebertaV3Preprocessor,
91
- )
92
- from keras_hub.src.models.deberta_v3.deberta_v3_tokenizer import (
93
- DebertaV3Tokenizer,
94
- )
95
- from keras_hub.src.models.deeplab_v3.deeplab_v3_backbone import (
96
- DeepLabV3Backbone,
97
- )
98
- from keras_hub.src.models.deeplab_v3.deeplab_v3_image_segmeter_preprocessor import (
99
- DeepLabV3ImageSegmenterPreprocessor,
100
- )
101
- from keras_hub.src.models.deeplab_v3.deeplab_v3_segmenter import (
102
- DeepLabV3ImageSegmenter,
103
- )
104
- from keras_hub.src.models.densenet.densenet_backbone import DenseNetBackbone
105
- from keras_hub.src.models.densenet.densenet_image_classifier import (
106
- DenseNetImageClassifier,
107
- )
108
- from keras_hub.src.models.densenet.densenet_image_classifier_preprocessor import (
109
- DenseNetImageClassifierPreprocessor,
110
- )
111
- from keras_hub.src.models.distil_bert.distil_bert_backbone import (
112
- DistilBertBackbone,
113
- )
114
- from keras_hub.src.models.distil_bert.distil_bert_masked_lm import (
115
- DistilBertMaskedLM,
116
- )
117
- from keras_hub.src.models.distil_bert.distil_bert_masked_lm_preprocessor import (
118
- DistilBertMaskedLMPreprocessor,
119
- )
120
- from keras_hub.src.models.distil_bert.distil_bert_text_classifier import (
121
- DistilBertTextClassifier,
122
- )
123
- from keras_hub.src.models.distil_bert.distil_bert_text_classifier import (
124
- DistilBertTextClassifier as DistilBertClassifier,
125
- )
126
- from keras_hub.src.models.distil_bert.distil_bert_text_classifier_preprocessor import (
127
- DistilBertTextClassifierPreprocessor,
128
- )
129
- from keras_hub.src.models.distil_bert.distil_bert_text_classifier_preprocessor import (
130
- DistilBertTextClassifierPreprocessor as DistilBertPreprocessor,
131
- )
132
- from keras_hub.src.models.distil_bert.distil_bert_tokenizer import (
133
- DistilBertTokenizer,
134
- )
135
- from keras_hub.src.models.efficientnet.efficientnet_backbone import (
136
- EfficientNetBackbone,
137
- )
138
- from keras_hub.src.models.efficientnet.efficientnet_image_classifier import (
139
- EfficientNetImageClassifier,
140
- )
141
- from keras_hub.src.models.efficientnet.efficientnet_image_classifier_preprocessor import (
142
- EfficientNetImageClassifierPreprocessor,
143
- )
144
- from keras_hub.src.models.electra.electra_backbone import ElectraBackbone
145
- from keras_hub.src.models.electra.electra_tokenizer import ElectraTokenizer
146
- from keras_hub.src.models.f_net.f_net_backbone import FNetBackbone
147
- from keras_hub.src.models.f_net.f_net_masked_lm import FNetMaskedLM
148
- from keras_hub.src.models.f_net.f_net_masked_lm_preprocessor import (
149
- FNetMaskedLMPreprocessor,
150
- )
151
- from keras_hub.src.models.f_net.f_net_text_classifier import FNetTextClassifier
152
- from keras_hub.src.models.f_net.f_net_text_classifier import (
153
- FNetTextClassifier as FNetClassifier,
154
- )
155
- from keras_hub.src.models.f_net.f_net_text_classifier_preprocessor import (
156
- FNetTextClassifierPreprocessor,
157
- )
158
- from keras_hub.src.models.f_net.f_net_text_classifier_preprocessor import (
159
- FNetTextClassifierPreprocessor as FNetPreprocessor,
160
- )
161
- from keras_hub.src.models.f_net.f_net_tokenizer import FNetTokenizer
162
- from keras_hub.src.models.falcon.falcon_backbone import FalconBackbone
163
- from keras_hub.src.models.falcon.falcon_causal_lm import FalconCausalLM
164
- from keras_hub.src.models.falcon.falcon_causal_lm_preprocessor import (
165
- FalconCausalLMPreprocessor,
166
- )
167
- from keras_hub.src.models.falcon.falcon_tokenizer import FalconTokenizer
168
- from keras_hub.src.models.feature_pyramid_backbone import FeaturePyramidBackbone
169
- from keras_hub.src.models.flux.flux_model import FluxBackbone
170
- from keras_hub.src.models.flux.flux_text_to_image import FluxTextToImage
171
- from keras_hub.src.models.flux.flux_text_to_image_preprocessor import (
172
- FluxTextToImagePreprocessor,
173
- )
174
- from keras_hub.src.models.gemma.gemma_backbone import GemmaBackbone
175
- from keras_hub.src.models.gemma.gemma_causal_lm import GemmaCausalLM
176
- from keras_hub.src.models.gemma.gemma_causal_lm_preprocessor import (
177
- GemmaCausalLMPreprocessor,
178
- )
179
- from keras_hub.src.models.gemma.gemma_tokenizer import GemmaTokenizer
180
- from keras_hub.src.models.gemma3.gemma3_backbone import Gemma3Backbone
181
- from keras_hub.src.models.gemma3.gemma3_causal_lm import Gemma3CausalLM
182
- from keras_hub.src.models.gemma3.gemma3_causal_lm_preprocessor import (
183
- Gemma3CausalLMPreprocessor,
184
- )
185
- from keras_hub.src.models.gemma3.gemma3_tokenizer import Gemma3Tokenizer
186
- from keras_hub.src.models.gemma3.gemma3_vision_encoder import (
187
- Gemma3VisionEncoder,
188
- )
189
- from keras_hub.src.models.gpt2.gpt2_backbone import GPT2Backbone
190
- from keras_hub.src.models.gpt2.gpt2_causal_lm import GPT2CausalLM
191
- from keras_hub.src.models.gpt2.gpt2_causal_lm_preprocessor import (
192
- GPT2CausalLMPreprocessor,
193
- )
194
- from keras_hub.src.models.gpt2.gpt2_preprocessor import GPT2Preprocessor
195
- from keras_hub.src.models.gpt2.gpt2_tokenizer import GPT2Tokenizer
196
- from keras_hub.src.models.gpt_neo_x.gpt_neo_x_backbone import GPTNeoXBackbone
197
- from keras_hub.src.models.gpt_neo_x.gpt_neo_x_causal_lm import GPTNeoXCausalLM
198
- from keras_hub.src.models.gpt_neo_x.gpt_neo_x_causal_lm_preprocessor import (
199
- GPTNeoXCausalLMPreprocessor,
200
- )
201
- from keras_hub.src.models.gpt_neo_x.gpt_neo_x_tokenizer import GPTNeoXTokenizer
202
- from keras_hub.src.models.image_classifier import ImageClassifier
203
- from keras_hub.src.models.image_classifier_preprocessor import (
204
- ImageClassifierPreprocessor,
205
- )
206
- from keras_hub.src.models.image_segmenter import ImageSegmenter
207
- from keras_hub.src.models.image_segmenter_preprocessor import (
208
- ImageSegmenterPreprocessor,
209
- )
210
- from keras_hub.src.models.image_to_image import ImageToImage
211
- from keras_hub.src.models.inpaint import Inpaint
212
- from keras_hub.src.models.llama.llama_backbone import LlamaBackbone
213
- from keras_hub.src.models.llama.llama_causal_lm import LlamaCausalLM
214
- from keras_hub.src.models.llama.llama_causal_lm_preprocessor import (
215
- LlamaCausalLMPreprocessor,
216
- )
217
- from keras_hub.src.models.llama.llama_tokenizer import LlamaTokenizer
218
- from keras_hub.src.models.llama3.llama3_backbone import Llama3Backbone
219
- from keras_hub.src.models.llama3.llama3_causal_lm import Llama3CausalLM
220
- from keras_hub.src.models.llama3.llama3_causal_lm_preprocessor import (
221
- Llama3CausalLMPreprocessor,
222
- )
223
- from keras_hub.src.models.llama3.llama3_tokenizer import Llama3Tokenizer
224
- from keras_hub.src.models.masked_lm import MaskedLM
225
- from keras_hub.src.models.masked_lm_preprocessor import MaskedLMPreprocessor
226
- from keras_hub.src.models.mistral.mistral_backbone import MistralBackbone
227
- from keras_hub.src.models.mistral.mistral_causal_lm import MistralCausalLM
228
- from keras_hub.src.models.mistral.mistral_causal_lm_preprocessor import (
229
- MistralCausalLMPreprocessor,
230
- )
231
- from keras_hub.src.models.mistral.mistral_tokenizer import MistralTokenizer
232
- from keras_hub.src.models.mit.mit_backbone import MiTBackbone
233
- from keras_hub.src.models.mit.mit_image_classifier import MiTImageClassifier
234
- from keras_hub.src.models.mit.mit_image_classifier_preprocessor import (
235
- MiTImageClassifierPreprocessor,
236
- )
237
- from keras_hub.src.models.mobilenet.mobilenet_backbone import MobileNetBackbone
238
- from keras_hub.src.models.mobilenet.mobilenet_image_classifier import (
239
- MobileNetImageClassifier,
240
- )
241
- from keras_hub.src.models.mobilenet.mobilenet_image_classifier_preprocessor import (
242
- MobileNetImageClassifierPreprocessor,
243
- )
244
- from keras_hub.src.models.object_detector import ObjectDetector
245
- from keras_hub.src.models.object_detector import (
246
- ObjectDetector as ImageObjectDetector,
247
- )
248
- from keras_hub.src.models.object_detector_preprocessor import (
249
- ObjectDetectorPreprocessor,
250
- )
251
- from keras_hub.src.models.object_detector_preprocessor import (
252
- ObjectDetectorPreprocessor as ImageObjectDetectorPreprocessor,
253
- )
254
- from keras_hub.src.models.opt.opt_backbone import OPTBackbone
255
- from keras_hub.src.models.opt.opt_causal_lm import OPTCausalLM
256
- from keras_hub.src.models.opt.opt_causal_lm_preprocessor import (
257
- OPTCausalLMPreprocessor,
258
- )
259
- from keras_hub.src.models.opt.opt_tokenizer import OPTTokenizer
260
- from keras_hub.src.models.pali_gemma.pali_gemma_backbone import (
261
- PaliGemmaBackbone,
262
- )
263
- from keras_hub.src.models.pali_gemma.pali_gemma_causal_lm import (
264
- PaliGemmaCausalLM,
265
- )
266
- from keras_hub.src.models.pali_gemma.pali_gemma_causal_lm_preprocessor import (
267
- PaliGemmaCausalLMPreprocessor,
268
- )
269
- from keras_hub.src.models.pali_gemma.pali_gemma_tokenizer import (
270
- PaliGemmaTokenizer,
271
- )
272
- from keras_hub.src.models.phi3.phi3_backbone import Phi3Backbone
273
- from keras_hub.src.models.phi3.phi3_causal_lm import Phi3CausalLM
274
- from keras_hub.src.models.phi3.phi3_causal_lm_preprocessor import (
275
- Phi3CausalLMPreprocessor,
276
- )
277
- from keras_hub.src.models.phi3.phi3_tokenizer import Phi3Tokenizer
278
- from keras_hub.src.models.preprocessor import Preprocessor
279
- from keras_hub.src.models.resnet.resnet_backbone import ResNetBackbone
280
- from keras_hub.src.models.resnet.resnet_image_classifier import (
281
- ResNetImageClassifier,
282
- )
283
- from keras_hub.src.models.resnet.resnet_image_classifier_preprocessor import (
284
- ResNetImageClassifierPreprocessor,
285
- )
286
- from keras_hub.src.models.retinanet.retinanet_backbone import RetinaNetBackbone
287
- from keras_hub.src.models.retinanet.retinanet_object_detector import (
288
- RetinaNetObjectDetector,
289
- )
290
- from keras_hub.src.models.retinanet.retinanet_object_detector_preprocessor import (
291
- RetinaNetObjectDetectorPreprocessor,
292
- )
293
- from keras_hub.src.models.roberta.roberta_backbone import RobertaBackbone
294
- from keras_hub.src.models.roberta.roberta_masked_lm import RobertaMaskedLM
295
- from keras_hub.src.models.roberta.roberta_masked_lm_preprocessor import (
296
- RobertaMaskedLMPreprocessor,
297
- )
298
- from keras_hub.src.models.roberta.roberta_text_classifier import (
299
- RobertaTextClassifier,
300
- )
301
- from keras_hub.src.models.roberta.roberta_text_classifier import (
302
- RobertaTextClassifier as RobertaClassifier,
303
- )
304
- from keras_hub.src.models.roberta.roberta_text_classifier_preprocessor import (
305
- RobertaTextClassifierPreprocessor,
306
- )
307
- from keras_hub.src.models.roberta.roberta_text_classifier_preprocessor import (
308
- RobertaTextClassifierPreprocessor as RobertaPreprocessor,
309
- )
310
- from keras_hub.src.models.roberta.roberta_tokenizer import RobertaTokenizer
311
- from keras_hub.src.models.roformer_v2.roformer_v2_backbone import (
312
- RoformerV2Backbone,
313
- )
314
- from keras_hub.src.models.roformer_v2.roformer_v2_masked_lm import (
315
- RoformerV2MaskedLM,
316
- )
317
- from keras_hub.src.models.roformer_v2.roformer_v2_masked_lm_preprocessor import (
318
- RoformerV2MaskedLMPreprocessor,
319
- )
320
- from keras_hub.src.models.roformer_v2.roformer_v2_text_classifier import (
321
- RoformerV2TextClassifier,
322
- )
323
- from keras_hub.src.models.roformer_v2.roformer_v2_text_classifier_preprocessor import (
324
- RoformerV2TextClassifierPreprocessor,
325
- )
326
- from keras_hub.src.models.roformer_v2.roformer_v2_tokenizer import (
327
- RoformerV2Tokenizer,
328
- )
329
- from keras_hub.src.models.sam.sam_backbone import SAMBackbone
330
- from keras_hub.src.models.sam.sam_image_segmenter import SAMImageSegmenter
331
- from keras_hub.src.models.sam.sam_image_segmenter_preprocessor import (
332
- SAMImageSegmenterPreprocessor,
333
- )
334
- from keras_hub.src.models.segformer.segformer_backbone import SegFormerBackbone
335
- from keras_hub.src.models.segformer.segformer_image_segmenter import (
336
- SegFormerImageSegmenter,
337
- )
338
- from keras_hub.src.models.segformer.segformer_image_segmenter_preprocessor import (
339
- SegFormerImageSegmenterPreprocessor,
340
- )
341
- from keras_hub.src.models.seq_2_seq_lm import Seq2SeqLM
342
- from keras_hub.src.models.seq_2_seq_lm_preprocessor import Seq2SeqLMPreprocessor
343
- from keras_hub.src.models.siglip.siglip_backbone import SigLIPBackbone
344
- from keras_hub.src.models.siglip.siglip_preprocessor import SigLIPPreprocessor
345
- from keras_hub.src.models.siglip.siglip_text_encoder import SigLIPTextEncoder
346
- from keras_hub.src.models.siglip.siglip_tokenizer import SigLIPTokenizer
347
- from keras_hub.src.models.siglip.siglip_vision_encoder import (
348
- SigLIPVisionEncoder,
349
- )
350
- from keras_hub.src.models.stable_diffusion_3.stable_diffusion_3_backbone import (
351
- StableDiffusion3Backbone,
352
- )
353
- from keras_hub.src.models.stable_diffusion_3.stable_diffusion_3_image_to_image import (
354
- StableDiffusion3ImageToImage,
355
- )
356
- from keras_hub.src.models.stable_diffusion_3.stable_diffusion_3_inpaint import (
357
- StableDiffusion3Inpaint,
358
- )
359
- from keras_hub.src.models.stable_diffusion_3.stable_diffusion_3_text_to_image import (
360
- StableDiffusion3TextToImage,
361
- )
362
- from keras_hub.src.models.stable_diffusion_3.stable_diffusion_3_text_to_image_preprocessor import (
363
- StableDiffusion3TextToImagePreprocessor,
364
- )
365
- from keras_hub.src.models.t5.t5_backbone import T5Backbone
366
- from keras_hub.src.models.t5.t5_preprocessor import T5Preprocessor
367
- from keras_hub.src.models.t5.t5_tokenizer import T5Tokenizer
368
- from keras_hub.src.models.task import Task
369
- from keras_hub.src.models.text_classifier import TextClassifier
370
- from keras_hub.src.models.text_classifier import TextClassifier as Classifier
371
- from keras_hub.src.models.text_classifier_preprocessor import (
372
- TextClassifierPreprocessor,
373
- )
374
- from keras_hub.src.models.text_to_image import TextToImage
375
- from keras_hub.src.models.text_to_image_preprocessor import (
376
- TextToImagePreprocessor,
377
- )
378
- from keras_hub.src.models.vgg.vgg_backbone import VGGBackbone
379
- from keras_hub.src.models.vgg.vgg_image_classifier import VGGImageClassifier
380
- from keras_hub.src.models.vgg.vgg_image_classifier_preprocessor import (
381
- VGGImageClassifierPreprocessor,
382
- )
383
- from keras_hub.src.models.vit.vit_backbone import ViTBackbone
384
- from keras_hub.src.models.vit.vit_image_classifier import ViTImageClassifier
385
- from keras_hub.src.models.vit.vit_image_classifier_preprocessor import (
386
- ViTImageClassifierPreprocessor,
387
- )
388
- from keras_hub.src.models.vit_det.vit_det_backbone import ViTDetBackbone
389
- from keras_hub.src.models.whisper.whisper_backbone import WhisperBackbone
390
- from keras_hub.src.models.whisper.whisper_tokenizer import WhisperTokenizer
391
- from keras_hub.src.models.xlm_roberta.xlm_roberta_backbone import (
392
- XLMRobertaBackbone,
393
- )
394
- from keras_hub.src.models.xlm_roberta.xlm_roberta_masked_lm import (
395
- XLMRobertaMaskedLM,
396
- )
397
- from keras_hub.src.models.xlm_roberta.xlm_roberta_masked_lm_preprocessor import (
398
- XLMRobertaMaskedLMPreprocessor,
399
- )
400
- from keras_hub.src.models.xlm_roberta.xlm_roberta_text_classifier import (
401
- XLMRobertaTextClassifier,
402
- )
403
- from keras_hub.src.models.xlm_roberta.xlm_roberta_text_classifier import (
404
- XLMRobertaTextClassifier as XLMRobertaClassifier,
405
- )
406
- from keras_hub.src.models.xlm_roberta.xlm_roberta_text_classifier_preprocessor import (
407
- XLMRobertaTextClassifierPreprocessor,
408
- )
409
- from keras_hub.src.models.xlm_roberta.xlm_roberta_text_classifier_preprocessor import (
410
- XLMRobertaTextClassifierPreprocessor as XLMRobertaPreprocessor,
411
- )
412
- from keras_hub.src.models.xlm_roberta.xlm_roberta_tokenizer import (
413
- XLMRobertaTokenizer,
414
- )
415
- from keras_hub.src.models.xlnet.xlnet_backbone import XLNetBackbone
416
- from keras_hub.src.tokenizers.tokenizer import Tokenizer
@@ -1,16 +0,0 @@
1
- """DO NOT EDIT.
2
-
3
- This file was autogenerated. Do not edit it by hand,
4
- since your modifications would be overwritten.
5
- """
6
-
7
- from keras_hub.src.samplers.beam_sampler import BeamSampler
8
- from keras_hub.src.samplers.contrastive_sampler import ContrastiveSampler
9
- from keras_hub.src.samplers.greedy_sampler import GreedySampler
10
- from keras_hub.src.samplers.random_sampler import RandomSampler
11
- from keras_hub.src.samplers.sampler import Sampler
12
- from keras_hub.src.samplers.serialization import deserialize
13
- from keras_hub.src.samplers.serialization import get
14
- from keras_hub.src.samplers.serialization import serialize
15
- from keras_hub.src.samplers.top_k_sampler import TopKSampler
16
- from keras_hub.src.samplers.top_p_sampler import TopPSampler
@@ -1,58 +0,0 @@
1
- """DO NOT EDIT.
2
-
3
- This file was autogenerated. Do not edit it by hand,
4
- since your modifications would be overwritten.
5
- """
6
-
7
- from keras_hub.src.models.albert.albert_tokenizer import AlbertTokenizer
8
- from keras_hub.src.models.bart.bart_tokenizer import BartTokenizer
9
- from keras_hub.src.models.bert.bert_tokenizer import BertTokenizer
10
- from keras_hub.src.models.bloom.bloom_tokenizer import BloomTokenizer
11
- from keras_hub.src.models.clip.clip_tokenizer import CLIPTokenizer
12
- from keras_hub.src.models.deberta_v3.deberta_v3_tokenizer import (
13
- DebertaV3Tokenizer,
14
- )
15
- from keras_hub.src.models.distil_bert.distil_bert_tokenizer import (
16
- DistilBertTokenizer,
17
- )
18
- from keras_hub.src.models.electra.electra_tokenizer import ElectraTokenizer
19
- from keras_hub.src.models.f_net.f_net_tokenizer import FNetTokenizer
20
- from keras_hub.src.models.falcon.falcon_tokenizer import FalconTokenizer
21
- from keras_hub.src.models.gemma.gemma_tokenizer import GemmaTokenizer
22
- from keras_hub.src.models.gemma3.gemma3_tokenizer import Gemma3Tokenizer
23
- from keras_hub.src.models.gpt2.gpt2_tokenizer import GPT2Tokenizer
24
- from keras_hub.src.models.gpt_neo_x.gpt_neo_x_tokenizer import GPTNeoXTokenizer
25
- from keras_hub.src.models.llama.llama_tokenizer import LlamaTokenizer
26
- from keras_hub.src.models.llama3.llama3_tokenizer import Llama3Tokenizer
27
- from keras_hub.src.models.mistral.mistral_tokenizer import MistralTokenizer
28
- from keras_hub.src.models.opt.opt_tokenizer import OPTTokenizer
29
- from keras_hub.src.models.pali_gemma.pali_gemma_tokenizer import (
30
- PaliGemmaTokenizer,
31
- )
32
- from keras_hub.src.models.phi3.phi3_tokenizer import Phi3Tokenizer
33
- from keras_hub.src.models.roberta.roberta_tokenizer import RobertaTokenizer
34
- from keras_hub.src.models.roformer_v2.roformer_v2_tokenizer import (
35
- RoformerV2Tokenizer,
36
- )
37
- from keras_hub.src.models.siglip.siglip_tokenizer import SigLIPTokenizer
38
- from keras_hub.src.models.t5.t5_tokenizer import T5Tokenizer
39
- from keras_hub.src.models.whisper.whisper_tokenizer import WhisperTokenizer
40
- from keras_hub.src.models.xlm_roberta.xlm_roberta_tokenizer import (
41
- XLMRobertaTokenizer,
42
- )
43
- from keras_hub.src.tokenizers.byte_pair_tokenizer import BytePairTokenizer
44
- from keras_hub.src.tokenizers.byte_tokenizer import ByteTokenizer
45
- from keras_hub.src.tokenizers.sentence_piece_tokenizer import (
46
- SentencePieceTokenizer,
47
- )
48
- from keras_hub.src.tokenizers.sentence_piece_tokenizer_trainer import (
49
- compute_sentence_piece_proto,
50
- )
51
- from keras_hub.src.tokenizers.tokenizer import Tokenizer
52
- from keras_hub.src.tokenizers.unicode_codepoint_tokenizer import (
53
- UnicodeCodepointTokenizer,
54
- )
55
- from keras_hub.src.tokenizers.word_piece_tokenizer import WordPieceTokenizer
56
- from keras_hub.src.tokenizers.word_piece_tokenizer_trainer import (
57
- compute_word_piece_vocabulary,
58
- )
@@ -1,9 +0,0 @@
1
- """DO NOT EDIT.
2
-
3
- This file was autogenerated. Do not edit it by hand,
4
- since your modifications would be overwritten.
5
- """
6
-
7
- from keras_hub.src.utils.imagenet.imagenet_utils import (
8
- decode_imagenet_predictions,
9
- )