__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,
)
from keras_hub.src.models.deepseek_v31.deepseek_v31_causal_lm_preprocessor import (
    DeepSeekV31CausalLMPreprocessor,
)
from keras_hub.src.models.deepseek_v31.deepseek_v31_presets import (
    backbone_presets,
    preprocessor_presets,
    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)