keras-hub-nightly 0.21.0.dev202504220402__py3-none-any.whl → 0.21.0.dev202504250023__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.
- keras_hub/__init__.py +15 -33
- keras_hub/layers/__init__.py +131 -0
- keras_hub/metrics/__init__.py +11 -0
- keras_hub/models/__init__.py +609 -0
- keras_hub/samplers/__init__.py +18 -0
- keras_hub/src/tests/mocks/mock_gemma3_tokenizer.py +155 -0
- keras_hub/src/utils/coco/__init__.py +0 -0
- keras_hub/src/utils/coco/coco_utils.py +133 -0
- keras_hub/src/utils/imagenet/imagenet_utils.py +36 -0
- keras_hub/src/utils/preset_utils.py +1 -1
- keras_hub/src/{version_utils.py → version.py} +1 -1
- keras_hub/tokenizers/__init__.py +108 -0
- keras_hub/utils/__init__.py +21 -0
- {keras_hub_nightly-0.21.0.dev202504220402.dist-info → keras_hub_nightly-0.21.0.dev202504250023.dist-info}/METADATA +6 -20
- {keras_hub_nightly-0.21.0.dev202504220402.dist-info → keras_hub_nightly-0.21.0.dev202504250023.dist-info}/RECORD +17 -15
- {keras_hub_nightly-0.21.0.dev202504220402.dist-info → keras_hub_nightly-0.21.0.dev202504250023.dist-info}/WHEEL +1 -1
- keras_hub/api/__init__.py +0 -15
- keras_hub/api/layers/__init__.py +0 -89
- keras_hub/api/metrics/__init__.py +0 -11
- keras_hub/api/models/__init__.py +0 -441
- keras_hub/api/samplers/__init__.py +0 -16
- keras_hub/api/tokenizers/__init__.py +0 -62
- keras_hub/api/utils/__init__.py +0 -9
- {keras_hub_nightly-0.21.0.dev202504220402.dist-info → keras_hub_nightly-0.21.0.dev202504250023.dist-info}/top_level.txt +0 -0
@@ -0,0 +1,609 @@
|
|
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 (
|
8
|
+
AlbertBackbone as AlbertBackbone,
|
9
|
+
)
|
10
|
+
from keras_hub.src.models.albert.albert_masked_lm import (
|
11
|
+
AlbertMaskedLM as AlbertMaskedLM,
|
12
|
+
)
|
13
|
+
from keras_hub.src.models.albert.albert_masked_lm_preprocessor import (
|
14
|
+
AlbertMaskedLMPreprocessor as AlbertMaskedLMPreprocessor,
|
15
|
+
)
|
16
|
+
from keras_hub.src.models.albert.albert_text_classifier import (
|
17
|
+
AlbertTextClassifier as AlbertClassifier,
|
18
|
+
)
|
19
|
+
from keras_hub.src.models.albert.albert_text_classifier import (
|
20
|
+
AlbertTextClassifier as AlbertTextClassifier,
|
21
|
+
)
|
22
|
+
from keras_hub.src.models.albert.albert_text_classifier_preprocessor import (
|
23
|
+
AlbertTextClassifierPreprocessor as AlbertPreprocessor,
|
24
|
+
)
|
25
|
+
from keras_hub.src.models.albert.albert_text_classifier_preprocessor import (
|
26
|
+
AlbertTextClassifierPreprocessor as AlbertTextClassifierPreprocessor,
|
27
|
+
)
|
28
|
+
from keras_hub.src.models.albert.albert_tokenizer import (
|
29
|
+
AlbertTokenizer as AlbertTokenizer,
|
30
|
+
)
|
31
|
+
from keras_hub.src.models.backbone import Backbone as Backbone
|
32
|
+
from keras_hub.src.models.bart.bart_backbone import BartBackbone as BartBackbone
|
33
|
+
from keras_hub.src.models.bart.bart_seq_2_seq_lm import (
|
34
|
+
BartSeq2SeqLM as BartSeq2SeqLM,
|
35
|
+
)
|
36
|
+
from keras_hub.src.models.bart.bart_seq_2_seq_lm_preprocessor import (
|
37
|
+
BartSeq2SeqLMPreprocessor as BartSeq2SeqLMPreprocessor,
|
38
|
+
)
|
39
|
+
from keras_hub.src.models.bart.bart_tokenizer import (
|
40
|
+
BartTokenizer as BartTokenizer,
|
41
|
+
)
|
42
|
+
from keras_hub.src.models.basnet.basnet import (
|
43
|
+
BASNetImageSegmenter as BASNetImageSegmenter,
|
44
|
+
)
|
45
|
+
from keras_hub.src.models.basnet.basnet_backbone import (
|
46
|
+
BASNetBackbone as BASNetBackbone,
|
47
|
+
)
|
48
|
+
from keras_hub.src.models.basnet.basnet_preprocessor import (
|
49
|
+
BASNetPreprocessor as BASNetPreprocessor,
|
50
|
+
)
|
51
|
+
from keras_hub.src.models.bert.bert_backbone import BertBackbone as BertBackbone
|
52
|
+
from keras_hub.src.models.bert.bert_masked_lm import (
|
53
|
+
BertMaskedLM as BertMaskedLM,
|
54
|
+
)
|
55
|
+
from keras_hub.src.models.bert.bert_masked_lm_preprocessor import (
|
56
|
+
BertMaskedLMPreprocessor as BertMaskedLMPreprocessor,
|
57
|
+
)
|
58
|
+
from keras_hub.src.models.bert.bert_text_classifier import (
|
59
|
+
BertTextClassifier as BertClassifier,
|
60
|
+
)
|
61
|
+
from keras_hub.src.models.bert.bert_text_classifier import (
|
62
|
+
BertTextClassifier as BertTextClassifier,
|
63
|
+
)
|
64
|
+
from keras_hub.src.models.bert.bert_text_classifier_preprocessor import (
|
65
|
+
BertTextClassifierPreprocessor as BertPreprocessor,
|
66
|
+
)
|
67
|
+
from keras_hub.src.models.bert.bert_text_classifier_preprocessor import (
|
68
|
+
BertTextClassifierPreprocessor as BertTextClassifierPreprocessor,
|
69
|
+
)
|
70
|
+
from keras_hub.src.models.bert.bert_tokenizer import (
|
71
|
+
BertTokenizer as BertTokenizer,
|
72
|
+
)
|
73
|
+
from keras_hub.src.models.bloom.bloom_backbone import (
|
74
|
+
BloomBackbone as BloomBackbone,
|
75
|
+
)
|
76
|
+
from keras_hub.src.models.bloom.bloom_causal_lm import (
|
77
|
+
BloomCausalLM as BloomCausalLM,
|
78
|
+
)
|
79
|
+
from keras_hub.src.models.bloom.bloom_causal_lm_preprocessor import (
|
80
|
+
BloomCausalLMPreprocessor as BloomCausalLMPreprocessor,
|
81
|
+
)
|
82
|
+
from keras_hub.src.models.bloom.bloom_tokenizer import (
|
83
|
+
BloomTokenizer as BloomTokenizer,
|
84
|
+
)
|
85
|
+
from keras_hub.src.models.causal_lm import CausalLM as CausalLM
|
86
|
+
from keras_hub.src.models.causal_lm_preprocessor import (
|
87
|
+
CausalLMPreprocessor as CausalLMPreprocessor,
|
88
|
+
)
|
89
|
+
from keras_hub.src.models.clip.clip_backbone import CLIPBackbone as CLIPBackbone
|
90
|
+
from keras_hub.src.models.clip.clip_preprocessor import (
|
91
|
+
CLIPPreprocessor as CLIPPreprocessor,
|
92
|
+
)
|
93
|
+
from keras_hub.src.models.clip.clip_text_encoder import (
|
94
|
+
CLIPTextEncoder as CLIPTextEncoder,
|
95
|
+
)
|
96
|
+
from keras_hub.src.models.clip.clip_tokenizer import (
|
97
|
+
CLIPTokenizer as CLIPTokenizer,
|
98
|
+
)
|
99
|
+
from keras_hub.src.models.clip.clip_vision_encoder import (
|
100
|
+
CLIPVisionEncoder as CLIPVisionEncoder,
|
101
|
+
)
|
102
|
+
from keras_hub.src.models.cspnet.cspnet_backbone import (
|
103
|
+
CSPNetBackbone as CSPNetBackbone,
|
104
|
+
)
|
105
|
+
from keras_hub.src.models.cspnet.cspnet_image_classifier import (
|
106
|
+
CSPNetImageClassifier as CSPNetImageClassifier,
|
107
|
+
)
|
108
|
+
from keras_hub.src.models.cspnet.cspnet_image_classifier_preprocessor import (
|
109
|
+
CSPNetImageClassifierPreprocessor as CSPNetImageClassifierPreprocessor,
|
110
|
+
)
|
111
|
+
from keras_hub.src.models.deberta_v3.deberta_v3_backbone import (
|
112
|
+
DebertaV3Backbone as DebertaV3Backbone,
|
113
|
+
)
|
114
|
+
from keras_hub.src.models.deberta_v3.deberta_v3_masked_lm import (
|
115
|
+
DebertaV3MaskedLM as DebertaV3MaskedLM,
|
116
|
+
)
|
117
|
+
from keras_hub.src.models.deberta_v3.deberta_v3_masked_lm_preprocessor import (
|
118
|
+
DebertaV3MaskedLMPreprocessor as DebertaV3MaskedLMPreprocessor,
|
119
|
+
)
|
120
|
+
from keras_hub.src.models.deberta_v3.deberta_v3_text_classifier import (
|
121
|
+
DebertaV3TextClassifier as DebertaV3Classifier,
|
122
|
+
)
|
123
|
+
from keras_hub.src.models.deberta_v3.deberta_v3_text_classifier import (
|
124
|
+
DebertaV3TextClassifier as DebertaV3TextClassifier,
|
125
|
+
)
|
126
|
+
from keras_hub.src.models.deberta_v3.deberta_v3_text_classifier_preprocessor import (
|
127
|
+
DebertaV3TextClassifierPreprocessor as DebertaV3Preprocessor,
|
128
|
+
)
|
129
|
+
from keras_hub.src.models.deberta_v3.deberta_v3_text_classifier_preprocessor import (
|
130
|
+
DebertaV3TextClassifierPreprocessor as DebertaV3TextClassifierPreprocessor,
|
131
|
+
)
|
132
|
+
from keras_hub.src.models.deberta_v3.deberta_v3_tokenizer import (
|
133
|
+
DebertaV3Tokenizer as DebertaV3Tokenizer,
|
134
|
+
)
|
135
|
+
from keras_hub.src.models.deeplab_v3.deeplab_v3_backbone import (
|
136
|
+
DeepLabV3Backbone as DeepLabV3Backbone,
|
137
|
+
)
|
138
|
+
from keras_hub.src.models.deeplab_v3.deeplab_v3_image_segmeter_preprocessor import (
|
139
|
+
DeepLabV3ImageSegmenterPreprocessor as DeepLabV3ImageSegmenterPreprocessor,
|
140
|
+
)
|
141
|
+
from keras_hub.src.models.deeplab_v3.deeplab_v3_segmenter import (
|
142
|
+
DeepLabV3ImageSegmenter as DeepLabV3ImageSegmenter,
|
143
|
+
)
|
144
|
+
from keras_hub.src.models.densenet.densenet_backbone import (
|
145
|
+
DenseNetBackbone as DenseNetBackbone,
|
146
|
+
)
|
147
|
+
from keras_hub.src.models.densenet.densenet_image_classifier import (
|
148
|
+
DenseNetImageClassifier as DenseNetImageClassifier,
|
149
|
+
)
|
150
|
+
from keras_hub.src.models.densenet.densenet_image_classifier_preprocessor import (
|
151
|
+
DenseNetImageClassifierPreprocessor as DenseNetImageClassifierPreprocessor,
|
152
|
+
)
|
153
|
+
from keras_hub.src.models.distil_bert.distil_bert_backbone import (
|
154
|
+
DistilBertBackbone as DistilBertBackbone,
|
155
|
+
)
|
156
|
+
from keras_hub.src.models.distil_bert.distil_bert_masked_lm import (
|
157
|
+
DistilBertMaskedLM as DistilBertMaskedLM,
|
158
|
+
)
|
159
|
+
from keras_hub.src.models.distil_bert.distil_bert_masked_lm_preprocessor import (
|
160
|
+
DistilBertMaskedLMPreprocessor as DistilBertMaskedLMPreprocessor,
|
161
|
+
)
|
162
|
+
from keras_hub.src.models.distil_bert.distil_bert_text_classifier import (
|
163
|
+
DistilBertTextClassifier as DistilBertClassifier,
|
164
|
+
)
|
165
|
+
from keras_hub.src.models.distil_bert.distil_bert_text_classifier import (
|
166
|
+
DistilBertTextClassifier as DistilBertTextClassifier,
|
167
|
+
)
|
168
|
+
from keras_hub.src.models.distil_bert.distil_bert_text_classifier_preprocessor import (
|
169
|
+
DistilBertTextClassifierPreprocessor as DistilBertPreprocessor,
|
170
|
+
)
|
171
|
+
from keras_hub.src.models.distil_bert.distil_bert_text_classifier_preprocessor import (
|
172
|
+
DistilBertTextClassifierPreprocessor as DistilBertTextClassifierPreprocessor,
|
173
|
+
)
|
174
|
+
from keras_hub.src.models.distil_bert.distil_bert_tokenizer import (
|
175
|
+
DistilBertTokenizer as DistilBertTokenizer,
|
176
|
+
)
|
177
|
+
from keras_hub.src.models.efficientnet.efficientnet_backbone import (
|
178
|
+
EfficientNetBackbone as EfficientNetBackbone,
|
179
|
+
)
|
180
|
+
from keras_hub.src.models.efficientnet.efficientnet_image_classifier import (
|
181
|
+
EfficientNetImageClassifier as EfficientNetImageClassifier,
|
182
|
+
)
|
183
|
+
from keras_hub.src.models.efficientnet.efficientnet_image_classifier_preprocessor import (
|
184
|
+
EfficientNetImageClassifierPreprocessor as EfficientNetImageClassifierPreprocessor,
|
185
|
+
)
|
186
|
+
from keras_hub.src.models.electra.electra_backbone import (
|
187
|
+
ElectraBackbone as ElectraBackbone,
|
188
|
+
)
|
189
|
+
from keras_hub.src.models.electra.electra_tokenizer import (
|
190
|
+
ElectraTokenizer as ElectraTokenizer,
|
191
|
+
)
|
192
|
+
from keras_hub.src.models.f_net.f_net_backbone import (
|
193
|
+
FNetBackbone as FNetBackbone,
|
194
|
+
)
|
195
|
+
from keras_hub.src.models.f_net.f_net_masked_lm import (
|
196
|
+
FNetMaskedLM as FNetMaskedLM,
|
197
|
+
)
|
198
|
+
from keras_hub.src.models.f_net.f_net_masked_lm_preprocessor import (
|
199
|
+
FNetMaskedLMPreprocessor as FNetMaskedLMPreprocessor,
|
200
|
+
)
|
201
|
+
from keras_hub.src.models.f_net.f_net_text_classifier import (
|
202
|
+
FNetTextClassifier as FNetClassifier,
|
203
|
+
)
|
204
|
+
from keras_hub.src.models.f_net.f_net_text_classifier import (
|
205
|
+
FNetTextClassifier as FNetTextClassifier,
|
206
|
+
)
|
207
|
+
from keras_hub.src.models.f_net.f_net_text_classifier_preprocessor import (
|
208
|
+
FNetTextClassifierPreprocessor as FNetPreprocessor,
|
209
|
+
)
|
210
|
+
from keras_hub.src.models.f_net.f_net_text_classifier_preprocessor import (
|
211
|
+
FNetTextClassifierPreprocessor as FNetTextClassifierPreprocessor,
|
212
|
+
)
|
213
|
+
from keras_hub.src.models.f_net.f_net_tokenizer import (
|
214
|
+
FNetTokenizer as FNetTokenizer,
|
215
|
+
)
|
216
|
+
from keras_hub.src.models.falcon.falcon_backbone import (
|
217
|
+
FalconBackbone as FalconBackbone,
|
218
|
+
)
|
219
|
+
from keras_hub.src.models.falcon.falcon_causal_lm import (
|
220
|
+
FalconCausalLM as FalconCausalLM,
|
221
|
+
)
|
222
|
+
from keras_hub.src.models.falcon.falcon_causal_lm_preprocessor import (
|
223
|
+
FalconCausalLMPreprocessor as FalconCausalLMPreprocessor,
|
224
|
+
)
|
225
|
+
from keras_hub.src.models.falcon.falcon_tokenizer import (
|
226
|
+
FalconTokenizer as FalconTokenizer,
|
227
|
+
)
|
228
|
+
from keras_hub.src.models.feature_pyramid_backbone import (
|
229
|
+
FeaturePyramidBackbone as FeaturePyramidBackbone,
|
230
|
+
)
|
231
|
+
from keras_hub.src.models.flux.flux_model import FluxBackbone as FluxBackbone
|
232
|
+
from keras_hub.src.models.flux.flux_text_to_image import (
|
233
|
+
FluxTextToImage as FluxTextToImage,
|
234
|
+
)
|
235
|
+
from keras_hub.src.models.flux.flux_text_to_image_preprocessor import (
|
236
|
+
FluxTextToImagePreprocessor as FluxTextToImagePreprocessor,
|
237
|
+
)
|
238
|
+
from keras_hub.src.models.gemma.gemma_backbone import (
|
239
|
+
GemmaBackbone as GemmaBackbone,
|
240
|
+
)
|
241
|
+
from keras_hub.src.models.gemma.gemma_causal_lm import (
|
242
|
+
GemmaCausalLM as GemmaCausalLM,
|
243
|
+
)
|
244
|
+
from keras_hub.src.models.gemma.gemma_causal_lm_preprocessor import (
|
245
|
+
GemmaCausalLMPreprocessor as GemmaCausalLMPreprocessor,
|
246
|
+
)
|
247
|
+
from keras_hub.src.models.gemma.gemma_tokenizer import (
|
248
|
+
GemmaTokenizer as GemmaTokenizer,
|
249
|
+
)
|
250
|
+
from keras_hub.src.models.gemma3.gemma3_backbone import (
|
251
|
+
Gemma3Backbone as Gemma3Backbone,
|
252
|
+
)
|
253
|
+
from keras_hub.src.models.gemma3.gemma3_causal_lm import (
|
254
|
+
Gemma3CausalLM as Gemma3CausalLM,
|
255
|
+
)
|
256
|
+
from keras_hub.src.models.gemma3.gemma3_causal_lm_preprocessor import (
|
257
|
+
Gemma3CausalLMPreprocessor as Gemma3CausalLMPreprocessor,
|
258
|
+
)
|
259
|
+
from keras_hub.src.models.gemma3.gemma3_tokenizer import (
|
260
|
+
Gemma3Tokenizer as Gemma3Tokenizer,
|
261
|
+
)
|
262
|
+
from keras_hub.src.models.gemma3.gemma3_vision_encoder import (
|
263
|
+
Gemma3VisionEncoder as Gemma3VisionEncoder,
|
264
|
+
)
|
265
|
+
from keras_hub.src.models.gpt2.gpt2_backbone import GPT2Backbone as GPT2Backbone
|
266
|
+
from keras_hub.src.models.gpt2.gpt2_causal_lm import (
|
267
|
+
GPT2CausalLM as GPT2CausalLM,
|
268
|
+
)
|
269
|
+
from keras_hub.src.models.gpt2.gpt2_causal_lm_preprocessor import (
|
270
|
+
GPT2CausalLMPreprocessor as GPT2CausalLMPreprocessor,
|
271
|
+
)
|
272
|
+
from keras_hub.src.models.gpt2.gpt2_preprocessor import (
|
273
|
+
GPT2Preprocessor as GPT2Preprocessor,
|
274
|
+
)
|
275
|
+
from keras_hub.src.models.gpt2.gpt2_tokenizer import (
|
276
|
+
GPT2Tokenizer as GPT2Tokenizer,
|
277
|
+
)
|
278
|
+
from keras_hub.src.models.gpt_neo_x.gpt_neo_x_backbone import (
|
279
|
+
GPTNeoXBackbone as GPTNeoXBackbone,
|
280
|
+
)
|
281
|
+
from keras_hub.src.models.gpt_neo_x.gpt_neo_x_causal_lm import (
|
282
|
+
GPTNeoXCausalLM as GPTNeoXCausalLM,
|
283
|
+
)
|
284
|
+
from keras_hub.src.models.gpt_neo_x.gpt_neo_x_causal_lm_preprocessor import (
|
285
|
+
GPTNeoXCausalLMPreprocessor as GPTNeoXCausalLMPreprocessor,
|
286
|
+
)
|
287
|
+
from keras_hub.src.models.gpt_neo_x.gpt_neo_x_tokenizer import (
|
288
|
+
GPTNeoXTokenizer as GPTNeoXTokenizer,
|
289
|
+
)
|
290
|
+
from keras_hub.src.models.image_classifier import (
|
291
|
+
ImageClassifier as ImageClassifier,
|
292
|
+
)
|
293
|
+
from keras_hub.src.models.image_classifier_preprocessor import (
|
294
|
+
ImageClassifierPreprocessor as ImageClassifierPreprocessor,
|
295
|
+
)
|
296
|
+
from keras_hub.src.models.image_segmenter import (
|
297
|
+
ImageSegmenter as ImageSegmenter,
|
298
|
+
)
|
299
|
+
from keras_hub.src.models.image_segmenter_preprocessor import (
|
300
|
+
ImageSegmenterPreprocessor as ImageSegmenterPreprocessor,
|
301
|
+
)
|
302
|
+
from keras_hub.src.models.image_to_image import ImageToImage as ImageToImage
|
303
|
+
from keras_hub.src.models.inpaint import Inpaint as Inpaint
|
304
|
+
from keras_hub.src.models.llama.llama_backbone import (
|
305
|
+
LlamaBackbone as LlamaBackbone,
|
306
|
+
)
|
307
|
+
from keras_hub.src.models.llama.llama_causal_lm import (
|
308
|
+
LlamaCausalLM as LlamaCausalLM,
|
309
|
+
)
|
310
|
+
from keras_hub.src.models.llama.llama_causal_lm_preprocessor import (
|
311
|
+
LlamaCausalLMPreprocessor as LlamaCausalLMPreprocessor,
|
312
|
+
)
|
313
|
+
from keras_hub.src.models.llama.llama_tokenizer import (
|
314
|
+
LlamaTokenizer as LlamaTokenizer,
|
315
|
+
)
|
316
|
+
from keras_hub.src.models.llama3.llama3_backbone import (
|
317
|
+
Llama3Backbone as Llama3Backbone,
|
318
|
+
)
|
319
|
+
from keras_hub.src.models.llama3.llama3_causal_lm import (
|
320
|
+
Llama3CausalLM as Llama3CausalLM,
|
321
|
+
)
|
322
|
+
from keras_hub.src.models.llama3.llama3_causal_lm_preprocessor import (
|
323
|
+
Llama3CausalLMPreprocessor as Llama3CausalLMPreprocessor,
|
324
|
+
)
|
325
|
+
from keras_hub.src.models.llama3.llama3_tokenizer import (
|
326
|
+
Llama3Tokenizer as Llama3Tokenizer,
|
327
|
+
)
|
328
|
+
from keras_hub.src.models.masked_lm import MaskedLM as MaskedLM
|
329
|
+
from keras_hub.src.models.masked_lm_preprocessor import (
|
330
|
+
MaskedLMPreprocessor as MaskedLMPreprocessor,
|
331
|
+
)
|
332
|
+
from keras_hub.src.models.mistral.mistral_backbone import (
|
333
|
+
MistralBackbone as MistralBackbone,
|
334
|
+
)
|
335
|
+
from keras_hub.src.models.mistral.mistral_causal_lm import (
|
336
|
+
MistralCausalLM as MistralCausalLM,
|
337
|
+
)
|
338
|
+
from keras_hub.src.models.mistral.mistral_causal_lm_preprocessor import (
|
339
|
+
MistralCausalLMPreprocessor as MistralCausalLMPreprocessor,
|
340
|
+
)
|
341
|
+
from keras_hub.src.models.mistral.mistral_tokenizer import (
|
342
|
+
MistralTokenizer as MistralTokenizer,
|
343
|
+
)
|
344
|
+
from keras_hub.src.models.mit.mit_backbone import MiTBackbone as MiTBackbone
|
345
|
+
from keras_hub.src.models.mit.mit_image_classifier import (
|
346
|
+
MiTImageClassifier as MiTImageClassifier,
|
347
|
+
)
|
348
|
+
from keras_hub.src.models.mit.mit_image_classifier_preprocessor import (
|
349
|
+
MiTImageClassifierPreprocessor as MiTImageClassifierPreprocessor,
|
350
|
+
)
|
351
|
+
from keras_hub.src.models.mobilenet.mobilenet_backbone import (
|
352
|
+
MobileNetBackbone as MobileNetBackbone,
|
353
|
+
)
|
354
|
+
from keras_hub.src.models.mobilenet.mobilenet_image_classifier import (
|
355
|
+
MobileNetImageClassifier as MobileNetImageClassifier,
|
356
|
+
)
|
357
|
+
from keras_hub.src.models.mobilenet.mobilenet_image_classifier_preprocessor import (
|
358
|
+
MobileNetImageClassifierPreprocessor as MobileNetImageClassifierPreprocessor,
|
359
|
+
)
|
360
|
+
from keras_hub.src.models.object_detector import (
|
361
|
+
ObjectDetector as ImageObjectDetector,
|
362
|
+
)
|
363
|
+
from keras_hub.src.models.object_detector import (
|
364
|
+
ObjectDetector as ObjectDetector,
|
365
|
+
)
|
366
|
+
from keras_hub.src.models.object_detector_preprocessor import (
|
367
|
+
ObjectDetectorPreprocessor as ImageObjectDetectorPreprocessor,
|
368
|
+
)
|
369
|
+
from keras_hub.src.models.object_detector_preprocessor import (
|
370
|
+
ObjectDetectorPreprocessor as ObjectDetectorPreprocessor,
|
371
|
+
)
|
372
|
+
from keras_hub.src.models.opt.opt_backbone import OPTBackbone as OPTBackbone
|
373
|
+
from keras_hub.src.models.opt.opt_causal_lm import OPTCausalLM as OPTCausalLM
|
374
|
+
from keras_hub.src.models.opt.opt_causal_lm_preprocessor import (
|
375
|
+
OPTCausalLMPreprocessor as OPTCausalLMPreprocessor,
|
376
|
+
)
|
377
|
+
from keras_hub.src.models.opt.opt_tokenizer import OPTTokenizer as OPTTokenizer
|
378
|
+
from keras_hub.src.models.pali_gemma.pali_gemma_backbone import (
|
379
|
+
PaliGemmaBackbone as PaliGemmaBackbone,
|
380
|
+
)
|
381
|
+
from keras_hub.src.models.pali_gemma.pali_gemma_causal_lm import (
|
382
|
+
PaliGemmaCausalLM as PaliGemmaCausalLM,
|
383
|
+
)
|
384
|
+
from keras_hub.src.models.pali_gemma.pali_gemma_causal_lm_preprocessor import (
|
385
|
+
PaliGemmaCausalLMPreprocessor as PaliGemmaCausalLMPreprocessor,
|
386
|
+
)
|
387
|
+
from keras_hub.src.models.pali_gemma.pali_gemma_tokenizer import (
|
388
|
+
PaliGemmaTokenizer as PaliGemmaTokenizer,
|
389
|
+
)
|
390
|
+
from keras_hub.src.models.phi3.phi3_backbone import Phi3Backbone as Phi3Backbone
|
391
|
+
from keras_hub.src.models.phi3.phi3_causal_lm import (
|
392
|
+
Phi3CausalLM as Phi3CausalLM,
|
393
|
+
)
|
394
|
+
from keras_hub.src.models.phi3.phi3_causal_lm_preprocessor import (
|
395
|
+
Phi3CausalLMPreprocessor as Phi3CausalLMPreprocessor,
|
396
|
+
)
|
397
|
+
from keras_hub.src.models.phi3.phi3_tokenizer import (
|
398
|
+
Phi3Tokenizer as Phi3Tokenizer,
|
399
|
+
)
|
400
|
+
from keras_hub.src.models.preprocessor import Preprocessor as Preprocessor
|
401
|
+
from keras_hub.src.models.qwen.qwen_backbone import (
|
402
|
+
QwenBackbone as Qwen2Backbone,
|
403
|
+
)
|
404
|
+
from keras_hub.src.models.qwen.qwen_backbone import QwenBackbone as QwenBackbone
|
405
|
+
from keras_hub.src.models.qwen.qwen_causal_lm import (
|
406
|
+
QwenCausalLM as Qwen2CausalLM,
|
407
|
+
)
|
408
|
+
from keras_hub.src.models.qwen.qwen_causal_lm import (
|
409
|
+
QwenCausalLM as QwenCausalLM,
|
410
|
+
)
|
411
|
+
from keras_hub.src.models.qwen.qwen_causal_lm_preprocessor import (
|
412
|
+
QwenCausalLMPreprocessor as Qwen2CausalLMPreprocessor,
|
413
|
+
)
|
414
|
+
from keras_hub.src.models.qwen.qwen_causal_lm_preprocessor import (
|
415
|
+
QwenCausalLMPreprocessor as QwenCausalLMPreprocessor,
|
416
|
+
)
|
417
|
+
from keras_hub.src.models.qwen.qwen_tokenizer import (
|
418
|
+
QwenTokenizer as Qwen2Tokenizer,
|
419
|
+
)
|
420
|
+
from keras_hub.src.models.qwen.qwen_tokenizer import (
|
421
|
+
QwenTokenizer as QwenTokenizer,
|
422
|
+
)
|
423
|
+
from keras_hub.src.models.resnet.resnet_backbone import (
|
424
|
+
ResNetBackbone as ResNetBackbone,
|
425
|
+
)
|
426
|
+
from keras_hub.src.models.resnet.resnet_image_classifier import (
|
427
|
+
ResNetImageClassifier as ResNetImageClassifier,
|
428
|
+
)
|
429
|
+
from keras_hub.src.models.resnet.resnet_image_classifier_preprocessor import (
|
430
|
+
ResNetImageClassifierPreprocessor as ResNetImageClassifierPreprocessor,
|
431
|
+
)
|
432
|
+
from keras_hub.src.models.retinanet.retinanet_backbone import (
|
433
|
+
RetinaNetBackbone as RetinaNetBackbone,
|
434
|
+
)
|
435
|
+
from keras_hub.src.models.retinanet.retinanet_object_detector import (
|
436
|
+
RetinaNetObjectDetector as RetinaNetObjectDetector,
|
437
|
+
)
|
438
|
+
from keras_hub.src.models.retinanet.retinanet_object_detector_preprocessor import (
|
439
|
+
RetinaNetObjectDetectorPreprocessor as RetinaNetObjectDetectorPreprocessor,
|
440
|
+
)
|
441
|
+
from keras_hub.src.models.roberta.roberta_backbone import (
|
442
|
+
RobertaBackbone as RobertaBackbone,
|
443
|
+
)
|
444
|
+
from keras_hub.src.models.roberta.roberta_masked_lm import (
|
445
|
+
RobertaMaskedLM as RobertaMaskedLM,
|
446
|
+
)
|
447
|
+
from keras_hub.src.models.roberta.roberta_masked_lm_preprocessor import (
|
448
|
+
RobertaMaskedLMPreprocessor as RobertaMaskedLMPreprocessor,
|
449
|
+
)
|
450
|
+
from keras_hub.src.models.roberta.roberta_text_classifier import (
|
451
|
+
RobertaTextClassifier as RobertaClassifier,
|
452
|
+
)
|
453
|
+
from keras_hub.src.models.roberta.roberta_text_classifier import (
|
454
|
+
RobertaTextClassifier as RobertaTextClassifier,
|
455
|
+
)
|
456
|
+
from keras_hub.src.models.roberta.roberta_text_classifier_preprocessor import (
|
457
|
+
RobertaTextClassifierPreprocessor as RobertaPreprocessor,
|
458
|
+
)
|
459
|
+
from keras_hub.src.models.roberta.roberta_text_classifier_preprocessor import (
|
460
|
+
RobertaTextClassifierPreprocessor as RobertaTextClassifierPreprocessor,
|
461
|
+
)
|
462
|
+
from keras_hub.src.models.roberta.roberta_tokenizer import (
|
463
|
+
RobertaTokenizer as RobertaTokenizer,
|
464
|
+
)
|
465
|
+
from keras_hub.src.models.roformer_v2.roformer_v2_backbone import (
|
466
|
+
RoformerV2Backbone as RoformerV2Backbone,
|
467
|
+
)
|
468
|
+
from keras_hub.src.models.roformer_v2.roformer_v2_masked_lm import (
|
469
|
+
RoformerV2MaskedLM as RoformerV2MaskedLM,
|
470
|
+
)
|
471
|
+
from keras_hub.src.models.roformer_v2.roformer_v2_masked_lm_preprocessor import (
|
472
|
+
RoformerV2MaskedLMPreprocessor as RoformerV2MaskedLMPreprocessor,
|
473
|
+
)
|
474
|
+
from keras_hub.src.models.roformer_v2.roformer_v2_text_classifier import (
|
475
|
+
RoformerV2TextClassifier as RoformerV2TextClassifier,
|
476
|
+
)
|
477
|
+
from keras_hub.src.models.roformer_v2.roformer_v2_text_classifier_preprocessor import (
|
478
|
+
RoformerV2TextClassifierPreprocessor as RoformerV2TextClassifierPreprocessor,
|
479
|
+
)
|
480
|
+
from keras_hub.src.models.roformer_v2.roformer_v2_tokenizer import (
|
481
|
+
RoformerV2Tokenizer as RoformerV2Tokenizer,
|
482
|
+
)
|
483
|
+
from keras_hub.src.models.sam.sam_backbone import SAMBackbone as SAMBackbone
|
484
|
+
from keras_hub.src.models.sam.sam_image_segmenter import (
|
485
|
+
SAMImageSegmenter as SAMImageSegmenter,
|
486
|
+
)
|
487
|
+
from keras_hub.src.models.sam.sam_image_segmenter_preprocessor import (
|
488
|
+
SAMImageSegmenterPreprocessor as SAMImageSegmenterPreprocessor,
|
489
|
+
)
|
490
|
+
from keras_hub.src.models.segformer.segformer_backbone import (
|
491
|
+
SegFormerBackbone as SegFormerBackbone,
|
492
|
+
)
|
493
|
+
from keras_hub.src.models.segformer.segformer_image_segmenter import (
|
494
|
+
SegFormerImageSegmenter as SegFormerImageSegmenter,
|
495
|
+
)
|
496
|
+
from keras_hub.src.models.segformer.segformer_image_segmenter_preprocessor import (
|
497
|
+
SegFormerImageSegmenterPreprocessor as SegFormerImageSegmenterPreprocessor,
|
498
|
+
)
|
499
|
+
from keras_hub.src.models.seq_2_seq_lm import Seq2SeqLM as Seq2SeqLM
|
500
|
+
from keras_hub.src.models.seq_2_seq_lm_preprocessor import (
|
501
|
+
Seq2SeqLMPreprocessor as Seq2SeqLMPreprocessor,
|
502
|
+
)
|
503
|
+
from keras_hub.src.models.siglip.siglip_backbone import (
|
504
|
+
SigLIPBackbone as SigLIPBackbone,
|
505
|
+
)
|
506
|
+
from keras_hub.src.models.siglip.siglip_preprocessor import (
|
507
|
+
SigLIPPreprocessor as SigLIPPreprocessor,
|
508
|
+
)
|
509
|
+
from keras_hub.src.models.siglip.siglip_text_encoder import (
|
510
|
+
SigLIPTextEncoder as SigLIPTextEncoder,
|
511
|
+
)
|
512
|
+
from keras_hub.src.models.siglip.siglip_tokenizer import (
|
513
|
+
SigLIPTokenizer as SigLIPTokenizer,
|
514
|
+
)
|
515
|
+
from keras_hub.src.models.siglip.siglip_vision_encoder import (
|
516
|
+
SigLIPVisionEncoder as SigLIPVisionEncoder,
|
517
|
+
)
|
518
|
+
from keras_hub.src.models.stable_diffusion_3.stable_diffusion_3_backbone import (
|
519
|
+
StableDiffusion3Backbone as StableDiffusion3Backbone,
|
520
|
+
)
|
521
|
+
from keras_hub.src.models.stable_diffusion_3.stable_diffusion_3_image_to_image import (
|
522
|
+
StableDiffusion3ImageToImage as StableDiffusion3ImageToImage,
|
523
|
+
)
|
524
|
+
from keras_hub.src.models.stable_diffusion_3.stable_diffusion_3_inpaint import (
|
525
|
+
StableDiffusion3Inpaint as StableDiffusion3Inpaint,
|
526
|
+
)
|
527
|
+
from keras_hub.src.models.stable_diffusion_3.stable_diffusion_3_text_to_image import (
|
528
|
+
StableDiffusion3TextToImage as StableDiffusion3TextToImage,
|
529
|
+
)
|
530
|
+
from keras_hub.src.models.stable_diffusion_3.stable_diffusion_3_text_to_image_preprocessor import (
|
531
|
+
StableDiffusion3TextToImagePreprocessor as StableDiffusion3TextToImagePreprocessor,
|
532
|
+
)
|
533
|
+
from keras_hub.src.models.t5.t5_backbone import T5Backbone as T5Backbone
|
534
|
+
from keras_hub.src.models.t5.t5_preprocessor import (
|
535
|
+
T5Preprocessor as T5Preprocessor,
|
536
|
+
)
|
537
|
+
from keras_hub.src.models.t5.t5_tokenizer import T5Tokenizer as T5Tokenizer
|
538
|
+
from keras_hub.src.models.task import Task as Task
|
539
|
+
from keras_hub.src.models.text_classifier import TextClassifier as Classifier
|
540
|
+
from keras_hub.src.models.text_classifier import (
|
541
|
+
TextClassifier as TextClassifier,
|
542
|
+
)
|
543
|
+
from keras_hub.src.models.text_classifier_preprocessor import (
|
544
|
+
TextClassifierPreprocessor as TextClassifierPreprocessor,
|
545
|
+
)
|
546
|
+
from keras_hub.src.models.text_to_image import TextToImage as TextToImage
|
547
|
+
from keras_hub.src.models.text_to_image_preprocessor import (
|
548
|
+
TextToImagePreprocessor as TextToImagePreprocessor,
|
549
|
+
)
|
550
|
+
from keras_hub.src.models.vgg.vgg_backbone import VGGBackbone as VGGBackbone
|
551
|
+
from keras_hub.src.models.vgg.vgg_image_classifier import (
|
552
|
+
VGGImageClassifier as VGGImageClassifier,
|
553
|
+
)
|
554
|
+
from keras_hub.src.models.vgg.vgg_image_classifier_preprocessor import (
|
555
|
+
VGGImageClassifierPreprocessor as VGGImageClassifierPreprocessor,
|
556
|
+
)
|
557
|
+
from keras_hub.src.models.vit.vit_backbone import ViTBackbone as ViTBackbone
|
558
|
+
from keras_hub.src.models.vit.vit_image_classifier import (
|
559
|
+
ViTImageClassifier as ViTImageClassifier,
|
560
|
+
)
|
561
|
+
from keras_hub.src.models.vit.vit_image_classifier_preprocessor import (
|
562
|
+
ViTImageClassifierPreprocessor as ViTImageClassifierPreprocessor,
|
563
|
+
)
|
564
|
+
from keras_hub.src.models.vit_det.vit_det_backbone import (
|
565
|
+
ViTDetBackbone as ViTDetBackbone,
|
566
|
+
)
|
567
|
+
from keras_hub.src.models.whisper.whisper_backbone import (
|
568
|
+
WhisperBackbone as WhisperBackbone,
|
569
|
+
)
|
570
|
+
from keras_hub.src.models.whisper.whisper_tokenizer import (
|
571
|
+
WhisperTokenizer as WhisperTokenizer,
|
572
|
+
)
|
573
|
+
from keras_hub.src.models.xception.xception_backbone import (
|
574
|
+
XceptionBackbone as XceptionBackbone,
|
575
|
+
)
|
576
|
+
from keras_hub.src.models.xception.xception_image_classifier import (
|
577
|
+
XceptionImageClassifier as XceptionImageClassifier,
|
578
|
+
)
|
579
|
+
from keras_hub.src.models.xception.xception_image_classifier_preprocessor import (
|
580
|
+
XceptionImageClassifierPreprocessor as XceptionImageClassifierPreprocessor,
|
581
|
+
)
|
582
|
+
from keras_hub.src.models.xlm_roberta.xlm_roberta_backbone import (
|
583
|
+
XLMRobertaBackbone as XLMRobertaBackbone,
|
584
|
+
)
|
585
|
+
from keras_hub.src.models.xlm_roberta.xlm_roberta_masked_lm import (
|
586
|
+
XLMRobertaMaskedLM as XLMRobertaMaskedLM,
|
587
|
+
)
|
588
|
+
from keras_hub.src.models.xlm_roberta.xlm_roberta_masked_lm_preprocessor import (
|
589
|
+
XLMRobertaMaskedLMPreprocessor as XLMRobertaMaskedLMPreprocessor,
|
590
|
+
)
|
591
|
+
from keras_hub.src.models.xlm_roberta.xlm_roberta_text_classifier import (
|
592
|
+
XLMRobertaTextClassifier as XLMRobertaClassifier,
|
593
|
+
)
|
594
|
+
from keras_hub.src.models.xlm_roberta.xlm_roberta_text_classifier import (
|
595
|
+
XLMRobertaTextClassifier as XLMRobertaTextClassifier,
|
596
|
+
)
|
597
|
+
from keras_hub.src.models.xlm_roberta.xlm_roberta_text_classifier_preprocessor import (
|
598
|
+
XLMRobertaTextClassifierPreprocessor as XLMRobertaPreprocessor,
|
599
|
+
)
|
600
|
+
from keras_hub.src.models.xlm_roberta.xlm_roberta_text_classifier_preprocessor import (
|
601
|
+
XLMRobertaTextClassifierPreprocessor as XLMRobertaTextClassifierPreprocessor,
|
602
|
+
)
|
603
|
+
from keras_hub.src.models.xlm_roberta.xlm_roberta_tokenizer import (
|
604
|
+
XLMRobertaTokenizer as XLMRobertaTokenizer,
|
605
|
+
)
|
606
|
+
from keras_hub.src.models.xlnet.xlnet_backbone import (
|
607
|
+
XLNetBackbone as XLNetBackbone,
|
608
|
+
)
|
609
|
+
from keras_hub.src.tokenizers.tokenizer import Tokenizer as Tokenizer
|
@@ -0,0 +1,18 @@
|
|
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 as BeamSampler
|
8
|
+
from keras_hub.src.samplers.contrastive_sampler import (
|
9
|
+
ContrastiveSampler as ContrastiveSampler,
|
10
|
+
)
|
11
|
+
from keras_hub.src.samplers.greedy_sampler import GreedySampler as GreedySampler
|
12
|
+
from keras_hub.src.samplers.random_sampler import RandomSampler as RandomSampler
|
13
|
+
from keras_hub.src.samplers.sampler import Sampler as Sampler
|
14
|
+
from keras_hub.src.samplers.serialization import deserialize as deserialize
|
15
|
+
from keras_hub.src.samplers.serialization import get as get
|
16
|
+
from keras_hub.src.samplers.serialization import serialize as serialize
|
17
|
+
from keras_hub.src.samplers.top_k_sampler import TopKSampler as TopKSampler
|
18
|
+
from keras_hub.src.samplers.top_p_sampler import TopPSampler as TopPSampler
|