__init__.py

"""DeepSeek V3.1 model exports."""

from keras_hub.src.models.deepseek_v31.deepseek_v31_backbone import (
    DeepSeekV31Backbone,
)
from keras_hub.src.models.deepseek_v31.deepseek_v31_causal_lm import (
    DeepSeekV31CausalLM,  # noqa: F401
)
from keras_hub.src.models.deepseek_v31.deepseek_v31_causal_lm_preprocessor import (  # noqa: E501
    DeepSeekV31CausalLMPreprocessor,
)
from keras_hub.src.models.deepseek_v31.deepseek_v31_presets import (
    backbone_presets,
)
from keras_hub.src.models.deepseek_v31.deepseek_v31_presets import (
    preprocessor_presets,
)
from keras_hub.src.models.deepseek_v31.deepseek_v31_presets import (
    tokenizer_presets,
)
from keras_hub.src.models.deepseek_v31.deepseek_v31_tokenizer import (
    DeepSeekV31Tokenizer,
)
from keras_hub.src.utils.preset_utils import register_presets

register_presets(backbone_presets, DeepSeekV31Backbone)
register_presets(tokenizer_presets, DeepSeekV31Tokenizer)
register_presets(preprocessor_presets, DeepSeekV31CausalLMPreprocessor)