Skip to content

Add Deepseek 3.1 & Deepseek 3.1 base #2607

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
cheachu wants to merge 10 commits into
base: master
Choose a base branch
from
Open

Add Deepseek 3.1 & Deepseek 3.1 base #2607

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
10 commits
Select commit Hold shift + click to select a range
c6145ae
Inital commit
cheachu Feb 21, 2026
2e811ac
added test files
cheachu Feb 21, 2026
88ce7a8
Merge branch 'keras-team:master' into deepseek-v3.1
cheachu Feb 21, 2026
21a02f2
chnaged naming convension and made sure it aligns with the contributi…
cheachu Feb 22, 2026
38348b2
Ruff fixes and added a fix for the tokenizer to handle sentencePiece…
cheachu Feb 22, 2026
2e1b4d8
some more ruff fixes
cheachu Feb 22, 2026
0ece165
Ruff patch
cheachu Feb 22, 2026
d04d045
Fixed the empty test case and dytpe issues and made sure all the ruf…
cheachu Feb 25, 2026
2b65b93
generate DeepSeek V3.1 public API files
cheachu Feb 25, 2026
102a080
added the import at the top
cheachu Feb 25, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 9 additions & 0 deletions keras_hub/api/models/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -150,6 +150,15 @@
from keras_hub.src.models.deeplab_v3.deeplab_v3_segmenter import (
DeepLabV3ImageSegmenter as DeepLabV3ImageSegmenter,
)
from keras_hub.src.models.deepseek_v31.deepseek_v31_backbone import (
DeepSeekV31Backbone as DeepSeekV31Backbone,
)
from keras_hub.src.models.deepseek_v31.deepseek_v31_causal_lm import (
DeepSeekV31CausalLM as DeepSeekV31CausalLM,
)
from keras_hub.src.models.deepseek_v31.deepseek_v31_causal_lm_preprocessor import (
DeepSeekV31CausalLMPreprocessor as DeepSeekV31CausalLMPreprocessor,
)
from keras_hub.src.models.deit.deit_backbone import DeiTBackbone as DeiTBackbone
from keras_hub.src.models.deit.deit_image_classifier import (
DeiTImageClassifier as DeiTImageClassifier,
Expand Down
3 changes: 3 additions & 0 deletions keras_hub/api/tokenizers/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -22,6 +22,9 @@
from keras_hub.src.models.deberta_v3.deberta_v3_tokenizer import (
DebertaV3Tokenizer as DebertaV3Tokenizer,
)
from keras_hub.src.models.deepseek_v31.deepseek_v31_tokenizer import (
DeepSeekV31Tokenizer as DeepSeekV31Tokenizer,
)
from keras_hub.src.models.distil_bert.distil_bert_tokenizer import (
DistilBertTokenizer as DistilBertTokenizer,
)
Expand Down
28 changes: 28 additions & 0 deletions keras_hub/src/models/deepseek_v31/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,28 @@
"""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)
Loading
Loading