ADD RWKV7

keras_hub/api/__init__.py

@@ -4,12 +4,12 @@
 since your modifications would be overwritten.
 """
 
-from keras_hub import layers as layers
-from keras_hub import metrics as metrics
-from keras_hub import models as models
-from keras_hub import samplers as samplers
-from keras_hub import tokenizers as tokenizers
-from keras_hub import utils as utils
-from keras_hub.src.utils.preset_utils import upload_preset as upload_preset
+from keras_hub import layers
+from keras_hub import metrics
+from keras_hub import models
+from keras_hub import samplers
+from keras_hub import tokenizers
+from keras_hub import utils
+from keras_hub.src.utils.preset_utils import upload_preset
 from keras_hub.src.version import __version__ as __version__
-from keras_hub.src.version import version as version
+from keras_hub.src.version import version

keras_hub/api/layers/__init__.py

@@ -4,149 +4,105 @@
 since your modifications would be overwritten.
 """
 
-from keras_hub.src.layers.modeling.alibi_bias import AlibiBias as AlibiBias
-from keras_hub.src.layers.modeling.anchor_generator import (
-    AnchorGenerator as AnchorGenerator,
-)
-from keras_hub.src.layers.modeling.box_matcher import BoxMatcher as BoxMatcher
+from keras_hub.src.layers.modeling.alibi_bias import AlibiBias
+from keras_hub.src.layers.modeling.anchor_generator import AnchorGenerator
+from keras_hub.src.layers.modeling.box_matcher import BoxMatcher
 from keras_hub.src.layers.modeling.cached_multi_head_attention import (
-    CachedMultiHeadAttention as CachedMultiHeadAttention,
-)
-from keras_hub.src.layers.modeling.f_net_encoder import (
-    FNetEncoder as FNetEncoder,
-)
-from keras_hub.src.layers.modeling.masked_lm_head import (
-    MaskedLMHead as MaskedLMHead,
-)
-from keras_hub.src.layers.modeling.non_max_supression import (
-    NonMaxSuppression as NonMaxSuppression,
-)
-from keras_hub.src.layers.modeling.position_embedding import (
-    PositionEmbedding as PositionEmbedding,
+    CachedMultiHeadAttention,
 )
+from keras_hub.src.layers.modeling.f_net_encoder import FNetEncoder
+from keras_hub.src.layers.modeling.masked_lm_head import MaskedLMHead
+from keras_hub.src.layers.modeling.non_max_supression import NonMaxSuppression
+from keras_hub.src.layers.modeling.position_embedding import PositionEmbedding
 from keras_hub.src.layers.modeling.reversible_embedding import (
-    ReversibleEmbedding as ReversibleEmbedding,
-)
-from keras_hub.src.layers.modeling.rms_normalization import (
-    RMSNormalization as RMSNormalization,
-)
-from keras_hub.src.layers.modeling.rotary_embedding import (
-    RotaryEmbedding as RotaryEmbedding,
+    ReversibleEmbedding,
 )
+from keras_hub.src.layers.modeling.rms_normalization import RMSNormalization
+from keras_hub.src.layers.modeling.rotary_embedding import RotaryEmbedding
 from keras_hub.src.layers.modeling.sine_position_encoding import (
-    SinePositionEncoding as SinePositionEncoding,
+    SinePositionEncoding,
 )
 from keras_hub.src.layers.modeling.token_and_position_embedding import (
-    TokenAndPositionEmbedding as TokenAndPositionEmbedding,
-)
-from keras_hub.src.layers.modeling.transformer_decoder import (
-    TransformerDecoder as TransformerDecoder,
-)
-from keras_hub.src.layers.modeling.transformer_encoder import (
-    TransformerEncoder as TransformerEncoder,
-)
-from keras_hub.src.layers.preprocessing.audio_converter import (
-    AudioConverter as AudioConverter,
-)
-from keras_hub.src.layers.preprocessing.image_converter import (
-    ImageConverter as ImageConverter,
+    TokenAndPositionEmbedding,
 )
+from keras_hub.src.layers.modeling.transformer_decoder import TransformerDecoder
+from keras_hub.src.layers.modeling.transformer_encoder import TransformerEncoder
+from keras_hub.src.layers.preprocessing.audio_converter import AudioConverter
+from keras_hub.src.layers.preprocessing.image_converter import ImageConverter
 from keras_hub.src.layers.preprocessing.masked_lm_mask_generator import (
-    MaskedLMMaskGenerator as MaskedLMMaskGenerator,
+    MaskedLMMaskGenerator,
 )
 from keras_hub.src.layers.preprocessing.multi_segment_packer import (
-    MultiSegmentPacker as MultiSegmentPacker,
-)
-from keras_hub.src.layers.preprocessing.random_deletion import (
-    RandomDeletion as RandomDeletion,
-)
-from keras_hub.src.layers.preprocessing.random_swap import (
-    RandomSwap as RandomSwap,
-)
-from keras_hub.src.layers.preprocessing.start_end_packer import (
-    StartEndPacker as StartEndPacker,
+    MultiSegmentPacker,
 )
+from keras_hub.src.layers.preprocessing.random_deletion import RandomDeletion
+from keras_hub.src.layers.preprocessing.random_swap import RandomSwap
+from keras_hub.src.layers.preprocessing.start_end_packer import StartEndPacker
 from keras_hub.src.models.basnet.basnet_image_converter import (
-    BASNetImageConverter as BASNetImageConverter,
-)
-from keras_hub.src.models.clip.clip_image_converter import (
-    CLIPImageConverter as CLIPImageConverter,
+    BASNetImageConverter,
 )
+from keras_hub.src.models.clip.clip_image_converter import CLIPImageConverter
 from keras_hub.src.models.cspnet.cspnet_image_converter import (
-    CSPNetImageConverter as CSPNetImageConverter,
+    CSPNetImageConverter,
 )
 from keras_hub.src.models.d_fine.d_fine_image_converter import (
-    DFineImageConverter as DFineImageConverter,
+    DFineImageConverter,
 )
 from keras_hub.src.models.deeplab_v3.deeplab_v3_image_converter import (
-    DeepLabV3ImageConverter as DeepLabV3ImageConverter,
-)
-from keras_hub.src.models.deit.deit_image_converter import (
-    DeiTImageConverter as DeiTImageConverter,
+    DeepLabV3ImageConverter,
 )
+from keras_hub.src.models.deit.deit_image_converter import DeiTImageConverter
 from keras_hub.src.models.densenet.densenet_image_converter import (
-    DenseNetImageConverter as DenseNetImageConverter,
+    DenseNetImageConverter,
 )
 from keras_hub.src.models.depth_anything.depth_anything_image_converter import (
-    DepthAnythingImageConverter as DepthAnythingImageConverter,
+    DepthAnythingImageConverter,
 )
 from keras_hub.src.models.dinov2.dinov2_image_converter import (
-    DINOV2ImageConverter as DINOV2ImageConverter,
+    DINOV2ImageConverter,
 )
 from keras_hub.src.models.efficientnet.efficientnet_image_converter import (
-    EfficientNetImageConverter as EfficientNetImageConverter,
+    EfficientNetImageConverter,
 )
 from keras_hub.src.models.gemma3.gemma3_image_converter import (
-    Gemma3ImageConverter as Gemma3ImageConverter,
+    Gemma3ImageConverter,
 )
 from keras_hub.src.models.hgnetv2.hgnetv2_image_converter import (
-    HGNetV2ImageConverter as HGNetV2ImageConverter,
-)
-from keras_hub.src.models.mit.mit_image_converter import (
-    MiTImageConverter as MiTImageConverter,
+    HGNetV2ImageConverter,
 )
+from keras_hub.src.models.mit.mit_image_converter import MiTImageConverter
 from keras_hub.src.models.mobilenet.mobilenet_image_converter import (
-    MobileNetImageConverter as MobileNetImageConverter,
+    MobileNetImageConverter,
 )
 from keras_hub.src.models.moonshine.moonshine_audio_converter import (
-    MoonshineAudioConverter as MoonshineAudioConverter,
+    MoonshineAudioConverter,
 )
 from keras_hub.src.models.pali_gemma.pali_gemma_image_converter import (
-    PaliGemmaImageConverter as PaliGemmaImageConverter,
+    PaliGemmaImageConverter,
 )
 from keras_hub.src.models.parseq.parseq_image_converter import (
-    PARSeqImageConverter as PARSeqImageConverter,
+    PARSeqImageConverter,
 )
 from keras_hub.src.models.resnet.resnet_image_converter import (
-    ResNetImageConverter as ResNetImageConverter,
+    ResNetImageConverter,
 )
 from keras_hub.src.models.retinanet.retinanet_image_converter import (
-    RetinaNetImageConverter as RetinaNetImageConverter,
-)
-from keras_hub.src.models.sam.sam_image_converter import (
-    SAMImageConverter as SAMImageConverter,
-)
-from keras_hub.src.models.sam.sam_mask_decoder import (
-    SAMMaskDecoder as SAMMaskDecoder,
-)
-from keras_hub.src.models.sam.sam_prompt_encoder import (
-    SAMPromptEncoder as SAMPromptEncoder,
+    RetinaNetImageConverter,
 )
+from keras_hub.src.models.sam.sam_image_converter import SAMImageConverter
+from keras_hub.src.models.sam.sam_mask_decoder import SAMMaskDecoder
+from keras_hub.src.models.sam.sam_prompt_encoder import SAMPromptEncoder
 from keras_hub.src.models.segformer.segformer_image_converter import (
-    SegFormerImageConverter as SegFormerImageConverter,
+    SegFormerImageConverter,
 )
 from keras_hub.src.models.siglip.siglip_image_converter import (
-    SigLIPImageConverter as SigLIPImageConverter,
-)
-from keras_hub.src.models.vgg.vgg_image_converter import (
-    VGGImageConverter as VGGImageConverter,
-)
-from keras_hub.src.models.vit.vit_image_converter import (
-    ViTImageConverter as ViTImageConverter,
+    SigLIPImageConverter,
 )
+from keras_hub.src.models.vgg.vgg_image_converter import VGGImageConverter
+from keras_hub.src.models.vit.vit_image_converter import ViTImageConverter
 from keras_hub.src.models.whisper.whisper_audio_converter import (
-    WhisperAudioConverter as WhisperAudioConverter,
+    WhisperAudioConverter,
 )
 from keras_hub.src.models.xception.xception_image_converter import (
-    XceptionImageConverter as XceptionImageConverter,
+    XceptionImageConverter,
 )

keras_hub/api/metrics/__init__.py

@@ -4,8 +4,8 @@
 since your modifications would be overwritten.
 """
 
-from keras_hub.src.metrics.bleu import Bleu as Bleu
-from keras_hub.src.metrics.edit_distance import EditDistance as EditDistance
-from keras_hub.src.metrics.perplexity import Perplexity as Perplexity
-from keras_hub.src.metrics.rouge_l import RougeL as RougeL
-from keras_hub.src.metrics.rouge_n import RougeN as RougeN
+from keras_hub.src.metrics.bleu import Bleu
+from keras_hub.src.metrics.edit_distance import EditDistance
+from keras_hub.src.metrics.perplexity import Perplexity
+from keras_hub.src.metrics.rouge_l import RougeL
+from keras_hub.src.metrics.rouge_n import RougeN