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