keras-hub-nightly 0.21.0.dev202504220402__py3-none-any.whl → 0.21.0.dev202504290404__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/models/stable_diffusion_3/stable_diffusion_3_backbone.py +6 -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.dev202504290404.dist-info}/METADATA +6 -20
- {keras_hub_nightly-0.21.0.dev202504220402.dist-info → keras_hub_nightly-0.21.0.dev202504290404.dist-info}/RECORD +18 -16
- {keras_hub_nightly-0.21.0.dev202504220402.dist-info → keras_hub_nightly-0.21.0.dev202504290404.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.dev202504290404.dist-info}/top_level.txt +0 -0
keras_hub/api/layers/__init__.py
DELETED
|
@@ -1,89 +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.layers.modeling.alibi_bias import AlibiBias
|
|
8
|
-
from keras_hub.src.layers.modeling.anchor_generator import AnchorGenerator
|
|
9
|
-
from keras_hub.src.layers.modeling.box_matcher import BoxMatcher
|
|
10
|
-
from keras_hub.src.layers.modeling.cached_multi_head_attention import (
|
|
11
|
-
CachedMultiHeadAttention,
|
|
12
|
-
)
|
|
13
|
-
from keras_hub.src.layers.modeling.f_net_encoder import FNetEncoder
|
|
14
|
-
from keras_hub.src.layers.modeling.masked_lm_head import MaskedLMHead
|
|
15
|
-
from keras_hub.src.layers.modeling.non_max_supression import NonMaxSuppression
|
|
16
|
-
from keras_hub.src.layers.modeling.position_embedding import PositionEmbedding
|
|
17
|
-
from keras_hub.src.layers.modeling.reversible_embedding import (
|
|
18
|
-
ReversibleEmbedding,
|
|
19
|
-
)
|
|
20
|
-
from keras_hub.src.layers.modeling.rms_normalization import RMSNormalization
|
|
21
|
-
from keras_hub.src.layers.modeling.rotary_embedding import RotaryEmbedding
|
|
22
|
-
from keras_hub.src.layers.modeling.sine_position_encoding import (
|
|
23
|
-
SinePositionEncoding,
|
|
24
|
-
)
|
|
25
|
-
from keras_hub.src.layers.modeling.token_and_position_embedding import (
|
|
26
|
-
TokenAndPositionEmbedding,
|
|
27
|
-
)
|
|
28
|
-
from keras_hub.src.layers.modeling.transformer_decoder import TransformerDecoder
|
|
29
|
-
from keras_hub.src.layers.modeling.transformer_encoder import TransformerEncoder
|
|
30
|
-
from keras_hub.src.layers.preprocessing.audio_converter import AudioConverter
|
|
31
|
-
from keras_hub.src.layers.preprocessing.image_converter import ImageConverter
|
|
32
|
-
from keras_hub.src.layers.preprocessing.masked_lm_mask_generator import (
|
|
33
|
-
MaskedLMMaskGenerator,
|
|
34
|
-
)
|
|
35
|
-
from keras_hub.src.layers.preprocessing.multi_segment_packer import (
|
|
36
|
-
MultiSegmentPacker,
|
|
37
|
-
)
|
|
38
|
-
from keras_hub.src.layers.preprocessing.random_deletion import RandomDeletion
|
|
39
|
-
from keras_hub.src.layers.preprocessing.random_swap import RandomSwap
|
|
40
|
-
from keras_hub.src.layers.preprocessing.start_end_packer import StartEndPacker
|
|
41
|
-
from keras_hub.src.models.basnet.basnet_image_converter import (
|
|
42
|
-
BASNetImageConverter,
|
|
43
|
-
)
|
|
44
|
-
from keras_hub.src.models.clip.clip_image_converter import CLIPImageConverter
|
|
45
|
-
from keras_hub.src.models.cspnet.cspnet_image_converter import (
|
|
46
|
-
CSPNetImageConverter,
|
|
47
|
-
)
|
|
48
|
-
from keras_hub.src.models.deeplab_v3.deeplab_v3_image_converter import (
|
|
49
|
-
DeepLabV3ImageConverter,
|
|
50
|
-
)
|
|
51
|
-
from keras_hub.src.models.densenet.densenet_image_converter import (
|
|
52
|
-
DenseNetImageConverter,
|
|
53
|
-
)
|
|
54
|
-
from keras_hub.src.models.efficientnet.efficientnet_image_converter import (
|
|
55
|
-
EfficientNetImageConverter,
|
|
56
|
-
)
|
|
57
|
-
from keras_hub.src.models.gemma3.gemma3_image_converter import (
|
|
58
|
-
Gemma3ImageConverter,
|
|
59
|
-
)
|
|
60
|
-
from keras_hub.src.models.mit.mit_image_converter import MiTImageConverter
|
|
61
|
-
from keras_hub.src.models.mobilenet.mobilenet_image_converter import (
|
|
62
|
-
MobileNetImageConverter,
|
|
63
|
-
)
|
|
64
|
-
from keras_hub.src.models.pali_gemma.pali_gemma_image_converter import (
|
|
65
|
-
PaliGemmaImageConverter,
|
|
66
|
-
)
|
|
67
|
-
from keras_hub.src.models.resnet.resnet_image_converter import (
|
|
68
|
-
ResNetImageConverter,
|
|
69
|
-
)
|
|
70
|
-
from keras_hub.src.models.retinanet.retinanet_image_converter import (
|
|
71
|
-
RetinaNetImageConverter,
|
|
72
|
-
)
|
|
73
|
-
from keras_hub.src.models.sam.sam_image_converter import SAMImageConverter
|
|
74
|
-
from keras_hub.src.models.sam.sam_mask_decoder import SAMMaskDecoder
|
|
75
|
-
from keras_hub.src.models.sam.sam_prompt_encoder import SAMPromptEncoder
|
|
76
|
-
from keras_hub.src.models.segformer.segformer_image_converter import (
|
|
77
|
-
SegFormerImageConverter,
|
|
78
|
-
)
|
|
79
|
-
from keras_hub.src.models.siglip.siglip_image_converter import (
|
|
80
|
-
SigLIPImageConverter,
|
|
81
|
-
)
|
|
82
|
-
from keras_hub.src.models.vgg.vgg_image_converter import VGGImageConverter
|
|
83
|
-
from keras_hub.src.models.vit.vit_image_converter import ViTImageConverter
|
|
84
|
-
from keras_hub.src.models.whisper.whisper_audio_converter import (
|
|
85
|
-
WhisperAudioConverter,
|
|
86
|
-
)
|
|
87
|
-
from keras_hub.src.models.xception.xception_image_converter import (
|
|
88
|
-
XceptionImageConverter,
|
|
89
|
-
)
|
|
@@ -1,11 +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.metrics.bleu import Bleu
|
|
8
|
-
from keras_hub.src.metrics.edit_distance import EditDistance
|
|
9
|
-
from keras_hub.src.metrics.perplexity import Perplexity
|
|
10
|
-
from keras_hub.src.metrics.rouge_l import RougeL
|
|
11
|
-
from keras_hub.src.metrics.rouge_n import RougeN
|
keras_hub/api/models/__init__.py
DELETED
|
@@ -1,441 +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.qwen.qwen_backbone import QwenBackbone
|
|
280
|
-
from keras_hub.src.models.qwen.qwen_backbone import (
|
|
281
|
-
QwenBackbone as Qwen2Backbone,
|
|
282
|
-
)
|
|
283
|
-
from keras_hub.src.models.qwen.qwen_causal_lm import QwenCausalLM
|
|
284
|
-
from keras_hub.src.models.qwen.qwen_causal_lm import (
|
|
285
|
-
QwenCausalLM as Qwen2CausalLM,
|
|
286
|
-
)
|
|
287
|
-
from keras_hub.src.models.qwen.qwen_causal_lm_preprocessor import (
|
|
288
|
-
QwenCausalLMPreprocessor,
|
|
289
|
-
)
|
|
290
|
-
from keras_hub.src.models.qwen.qwen_causal_lm_preprocessor import (
|
|
291
|
-
QwenCausalLMPreprocessor as Qwen2CausalLMPreprocessor,
|
|
292
|
-
)
|
|
293
|
-
from keras_hub.src.models.qwen.qwen_tokenizer import QwenTokenizer
|
|
294
|
-
from keras_hub.src.models.qwen.qwen_tokenizer import (
|
|
295
|
-
QwenTokenizer as Qwen2Tokenizer,
|
|
296
|
-
)
|
|
297
|
-
from keras_hub.src.models.resnet.resnet_backbone import ResNetBackbone
|
|
298
|
-
from keras_hub.src.models.resnet.resnet_image_classifier import (
|
|
299
|
-
ResNetImageClassifier,
|
|
300
|
-
)
|
|
301
|
-
from keras_hub.src.models.resnet.resnet_image_classifier_preprocessor import (
|
|
302
|
-
ResNetImageClassifierPreprocessor,
|
|
303
|
-
)
|
|
304
|
-
from keras_hub.src.models.retinanet.retinanet_backbone import RetinaNetBackbone
|
|
305
|
-
from keras_hub.src.models.retinanet.retinanet_object_detector import (
|
|
306
|
-
RetinaNetObjectDetector,
|
|
307
|
-
)
|
|
308
|
-
from keras_hub.src.models.retinanet.retinanet_object_detector_preprocessor import (
|
|
309
|
-
RetinaNetObjectDetectorPreprocessor,
|
|
310
|
-
)
|
|
311
|
-
from keras_hub.src.models.roberta.roberta_backbone import RobertaBackbone
|
|
312
|
-
from keras_hub.src.models.roberta.roberta_masked_lm import RobertaMaskedLM
|
|
313
|
-
from keras_hub.src.models.roberta.roberta_masked_lm_preprocessor import (
|
|
314
|
-
RobertaMaskedLMPreprocessor,
|
|
315
|
-
)
|
|
316
|
-
from keras_hub.src.models.roberta.roberta_text_classifier import (
|
|
317
|
-
RobertaTextClassifier,
|
|
318
|
-
)
|
|
319
|
-
from keras_hub.src.models.roberta.roberta_text_classifier import (
|
|
320
|
-
RobertaTextClassifier as RobertaClassifier,
|
|
321
|
-
)
|
|
322
|
-
from keras_hub.src.models.roberta.roberta_text_classifier_preprocessor import (
|
|
323
|
-
RobertaTextClassifierPreprocessor,
|
|
324
|
-
)
|
|
325
|
-
from keras_hub.src.models.roberta.roberta_text_classifier_preprocessor import (
|
|
326
|
-
RobertaTextClassifierPreprocessor as RobertaPreprocessor,
|
|
327
|
-
)
|
|
328
|
-
from keras_hub.src.models.roberta.roberta_tokenizer import RobertaTokenizer
|
|
329
|
-
from keras_hub.src.models.roformer_v2.roformer_v2_backbone import (
|
|
330
|
-
RoformerV2Backbone,
|
|
331
|
-
)
|
|
332
|
-
from keras_hub.src.models.roformer_v2.roformer_v2_masked_lm import (
|
|
333
|
-
RoformerV2MaskedLM,
|
|
334
|
-
)
|
|
335
|
-
from keras_hub.src.models.roformer_v2.roformer_v2_masked_lm_preprocessor import (
|
|
336
|
-
RoformerV2MaskedLMPreprocessor,
|
|
337
|
-
)
|
|
338
|
-
from keras_hub.src.models.roformer_v2.roformer_v2_text_classifier import (
|
|
339
|
-
RoformerV2TextClassifier,
|
|
340
|
-
)
|
|
341
|
-
from keras_hub.src.models.roformer_v2.roformer_v2_text_classifier_preprocessor import (
|
|
342
|
-
RoformerV2TextClassifierPreprocessor,
|
|
343
|
-
)
|
|
344
|
-
from keras_hub.src.models.roformer_v2.roformer_v2_tokenizer import (
|
|
345
|
-
RoformerV2Tokenizer,
|
|
346
|
-
)
|
|
347
|
-
from keras_hub.src.models.sam.sam_backbone import SAMBackbone
|
|
348
|
-
from keras_hub.src.models.sam.sam_image_segmenter import SAMImageSegmenter
|
|
349
|
-
from keras_hub.src.models.sam.sam_image_segmenter_preprocessor import (
|
|
350
|
-
SAMImageSegmenterPreprocessor,
|
|
351
|
-
)
|
|
352
|
-
from keras_hub.src.models.segformer.segformer_backbone import SegFormerBackbone
|
|
353
|
-
from keras_hub.src.models.segformer.segformer_image_segmenter import (
|
|
354
|
-
SegFormerImageSegmenter,
|
|
355
|
-
)
|
|
356
|
-
from keras_hub.src.models.segformer.segformer_image_segmenter_preprocessor import (
|
|
357
|
-
SegFormerImageSegmenterPreprocessor,
|
|
358
|
-
)
|
|
359
|
-
from keras_hub.src.models.seq_2_seq_lm import Seq2SeqLM
|
|
360
|
-
from keras_hub.src.models.seq_2_seq_lm_preprocessor import Seq2SeqLMPreprocessor
|
|
361
|
-
from keras_hub.src.models.siglip.siglip_backbone import SigLIPBackbone
|
|
362
|
-
from keras_hub.src.models.siglip.siglip_preprocessor import SigLIPPreprocessor
|
|
363
|
-
from keras_hub.src.models.siglip.siglip_text_encoder import SigLIPTextEncoder
|
|
364
|
-
from keras_hub.src.models.siglip.siglip_tokenizer import SigLIPTokenizer
|
|
365
|
-
from keras_hub.src.models.siglip.siglip_vision_encoder import (
|
|
366
|
-
SigLIPVisionEncoder,
|
|
367
|
-
)
|
|
368
|
-
from keras_hub.src.models.stable_diffusion_3.stable_diffusion_3_backbone import (
|
|
369
|
-
StableDiffusion3Backbone,
|
|
370
|
-
)
|
|
371
|
-
from keras_hub.src.models.stable_diffusion_3.stable_diffusion_3_image_to_image import (
|
|
372
|
-
StableDiffusion3ImageToImage,
|
|
373
|
-
)
|
|
374
|
-
from keras_hub.src.models.stable_diffusion_3.stable_diffusion_3_inpaint import (
|
|
375
|
-
StableDiffusion3Inpaint,
|
|
376
|
-
)
|
|
377
|
-
from keras_hub.src.models.stable_diffusion_3.stable_diffusion_3_text_to_image import (
|
|
378
|
-
StableDiffusion3TextToImage,
|
|
379
|
-
)
|
|
380
|
-
from keras_hub.src.models.stable_diffusion_3.stable_diffusion_3_text_to_image_preprocessor import (
|
|
381
|
-
StableDiffusion3TextToImagePreprocessor,
|
|
382
|
-
)
|
|
383
|
-
from keras_hub.src.models.t5.t5_backbone import T5Backbone
|
|
384
|
-
from keras_hub.src.models.t5.t5_preprocessor import T5Preprocessor
|
|
385
|
-
from keras_hub.src.models.t5.t5_tokenizer import T5Tokenizer
|
|
386
|
-
from keras_hub.src.models.task import Task
|
|
387
|
-
from keras_hub.src.models.text_classifier import TextClassifier
|
|
388
|
-
from keras_hub.src.models.text_classifier import TextClassifier as Classifier
|
|
389
|
-
from keras_hub.src.models.text_classifier_preprocessor import (
|
|
390
|
-
TextClassifierPreprocessor,
|
|
391
|
-
)
|
|
392
|
-
from keras_hub.src.models.text_to_image import TextToImage
|
|
393
|
-
from keras_hub.src.models.text_to_image_preprocessor import (
|
|
394
|
-
TextToImagePreprocessor,
|
|
395
|
-
)
|
|
396
|
-
from keras_hub.src.models.vgg.vgg_backbone import VGGBackbone
|
|
397
|
-
from keras_hub.src.models.vgg.vgg_image_classifier import VGGImageClassifier
|
|
398
|
-
from keras_hub.src.models.vgg.vgg_image_classifier_preprocessor import (
|
|
399
|
-
VGGImageClassifierPreprocessor,
|
|
400
|
-
)
|
|
401
|
-
from keras_hub.src.models.vit.vit_backbone import ViTBackbone
|
|
402
|
-
from keras_hub.src.models.vit.vit_image_classifier import ViTImageClassifier
|
|
403
|
-
from keras_hub.src.models.vit.vit_image_classifier_preprocessor import (
|
|
404
|
-
ViTImageClassifierPreprocessor,
|
|
405
|
-
)
|
|
406
|
-
from keras_hub.src.models.vit_det.vit_det_backbone import ViTDetBackbone
|
|
407
|
-
from keras_hub.src.models.whisper.whisper_backbone import WhisperBackbone
|
|
408
|
-
from keras_hub.src.models.whisper.whisper_tokenizer import WhisperTokenizer
|
|
409
|
-
from keras_hub.src.models.xception.xception_backbone import XceptionBackbone
|
|
410
|
-
from keras_hub.src.models.xception.xception_image_classifier import (
|
|
411
|
-
XceptionImageClassifier,
|
|
412
|
-
)
|
|
413
|
-
from keras_hub.src.models.xception.xception_image_classifier_preprocessor import (
|
|
414
|
-
XceptionImageClassifierPreprocessor,
|
|
415
|
-
)
|
|
416
|
-
from keras_hub.src.models.xlm_roberta.xlm_roberta_backbone import (
|
|
417
|
-
XLMRobertaBackbone,
|
|
418
|
-
)
|
|
419
|
-
from keras_hub.src.models.xlm_roberta.xlm_roberta_masked_lm import (
|
|
420
|
-
XLMRobertaMaskedLM,
|
|
421
|
-
)
|
|
422
|
-
from keras_hub.src.models.xlm_roberta.xlm_roberta_masked_lm_preprocessor import (
|
|
423
|
-
XLMRobertaMaskedLMPreprocessor,
|
|
424
|
-
)
|
|
425
|
-
from keras_hub.src.models.xlm_roberta.xlm_roberta_text_classifier import (
|
|
426
|
-
XLMRobertaTextClassifier,
|
|
427
|
-
)
|
|
428
|
-
from keras_hub.src.models.xlm_roberta.xlm_roberta_text_classifier import (
|
|
429
|
-
XLMRobertaTextClassifier as XLMRobertaClassifier,
|
|
430
|
-
)
|
|
431
|
-
from keras_hub.src.models.xlm_roberta.xlm_roberta_text_classifier_preprocessor import (
|
|
432
|
-
XLMRobertaTextClassifierPreprocessor,
|
|
433
|
-
)
|
|
434
|
-
from keras_hub.src.models.xlm_roberta.xlm_roberta_text_classifier_preprocessor import (
|
|
435
|
-
XLMRobertaTextClassifierPreprocessor as XLMRobertaPreprocessor,
|
|
436
|
-
)
|
|
437
|
-
from keras_hub.src.models.xlm_roberta.xlm_roberta_tokenizer import (
|
|
438
|
-
XLMRobertaTokenizer,
|
|
439
|
-
)
|
|
440
|
-
from keras_hub.src.models.xlnet.xlnet_backbone import XLNetBackbone
|
|
441
|
-
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,62 +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.qwen.qwen_tokenizer import QwenTokenizer
|
|
34
|
-
from keras_hub.src.models.qwen.qwen_tokenizer import (
|
|
35
|
-
QwenTokenizer as Qwen2Tokenizer,
|
|
36
|
-
)
|
|
37
|
-
from keras_hub.src.models.roberta.roberta_tokenizer import RobertaTokenizer
|
|
38
|
-
from keras_hub.src.models.roformer_v2.roformer_v2_tokenizer import (
|
|
39
|
-
RoformerV2Tokenizer,
|
|
40
|
-
)
|
|
41
|
-
from keras_hub.src.models.siglip.siglip_tokenizer import SigLIPTokenizer
|
|
42
|
-
from keras_hub.src.models.t5.t5_tokenizer import T5Tokenizer
|
|
43
|
-
from keras_hub.src.models.whisper.whisper_tokenizer import WhisperTokenizer
|
|
44
|
-
from keras_hub.src.models.xlm_roberta.xlm_roberta_tokenizer import (
|
|
45
|
-
XLMRobertaTokenizer,
|
|
46
|
-
)
|
|
47
|
-
from keras_hub.src.tokenizers.byte_pair_tokenizer import BytePairTokenizer
|
|
48
|
-
from keras_hub.src.tokenizers.byte_tokenizer import ByteTokenizer
|
|
49
|
-
from keras_hub.src.tokenizers.sentence_piece_tokenizer import (
|
|
50
|
-
SentencePieceTokenizer,
|
|
51
|
-
)
|
|
52
|
-
from keras_hub.src.tokenizers.sentence_piece_tokenizer_trainer import (
|
|
53
|
-
compute_sentence_piece_proto,
|
|
54
|
-
)
|
|
55
|
-
from keras_hub.src.tokenizers.tokenizer import Tokenizer
|
|
56
|
-
from keras_hub.src.tokenizers.unicode_codepoint_tokenizer import (
|
|
57
|
-
UnicodeCodepointTokenizer,
|
|
58
|
-
)
|
|
59
|
-
from keras_hub.src.tokenizers.word_piece_tokenizer import WordPieceTokenizer
|
|
60
|
-
from keras_hub.src.tokenizers.word_piece_tokenizer_trainer import (
|
|
61
|
-
compute_word_piece_vocabulary,
|
|
62
|
-
)
|
keras_hub/api/utils/__init__.py
DELETED