Skip to content

Commit b1a2950

Browse files
committed
add deepseek to docs
1 parent 6dc02fd commit b1a2950

File tree

3 files changed

+6
-5
lines changed

3 files changed

+6
-5
lines changed

docs/source/openvino/models.mdx

+3
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,9 @@ Here is the list of the supported architectures :
4343
- Deberta-v2
4444
- DeciLM
4545
- Deit
46+
- Deepseek
47+
- Deepseek_v2
48+
- Deepseek_v3
4649
- DistilBert
4750
- Electra
4851
- Encoder Decoder

tests/openvino/test_modeling.py

+2-3
Original file line numberDiff line numberDiff line change
@@ -995,7 +995,7 @@ class OVModelForCausalLMIntegrationTest(unittest.TestCase):
995995

996996
# openvino 2025.0 required for disabling check_trace
997997
if is_openvino_version(">=", "2025.0"):
998-
SUPPORTED_ARCHITECTURES += ("deepseek-v2", "deepseek-v3")
998+
SUPPORTED_ARCHITECTURES += "deepseek"
999999

10001000
# gptq and awq install disabled for windows test environment
10011001
if platform.system() != "Windows":
@@ -1025,8 +1025,7 @@ class OVModelForCausalLMIntegrationTest(unittest.TestCase):
10251025
"exaone",
10261026
"decilm",
10271027
"minicpm3",
1028-
"deepseek-v2",
1029-
"deepseek-v3",
1028+
"deepseek",
10301029
)
10311030

10321031
@parameterized.expand(SUPPORTED_ARCHITECTURES)

tests/openvino/utils_tests.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -51,8 +51,7 @@
5151
"deberta": "hf-internal-testing/tiny-random-deberta",
5252
"deberta_v2": "hf-internal-testing/tiny-random-DebertaV2Model",
5353
"decilm": "katuni4ka/tiny-random-decilm",
54-
"deepseek-v3": "katuni4ka/tiny-random-deepseek-v3",
55-
"deepseek-v2": "yujiepan/deepseek-v2-tiny-random",
54+
"deepseek": "yujiepan/deepseek-v2-tiny-random",
5655
"deit": "hf-internal-testing/tiny-random-DeiTModel",
5756
"convnext": "hf-internal-testing/tiny-random-convnext",
5857
"convnextv2": "hf-internal-testing/tiny-random-ConvNextV2Model",

0 commit comments

Comments
 (0)