File tree 3 files changed +6
-5
lines changed
3 files changed +6
-5
lines changed Original file line number Diff line number Diff line change @@ -43,6 +43,9 @@ Here is the list of the supported architectures :
43
43
- Deberta-v2
44
44
- DeciLM
45
45
- Deit
46
+ - Deepseek
47
+ - Deepseek_v2
48
+ - Deepseek_v3
46
49
- DistilBert
47
50
- Electra
48
51
- Encoder Decoder
Original file line number Diff line number Diff line change @@ -995,7 +995,7 @@ class OVModelForCausalLMIntegrationTest(unittest.TestCase):
995
995
996
996
# openvino 2025.0 required for disabling check_trace
997
997
if is_openvino_version (">=" , "2025.0" ):
998
- SUPPORTED_ARCHITECTURES += ( "deepseek-v2" , "deepseek-v3" )
998
+ SUPPORTED_ARCHITECTURES += "deepseek"
999
999
1000
1000
# gptq and awq install disabled for windows test environment
1001
1001
if platform .system () != "Windows" :
@@ -1025,8 +1025,7 @@ class OVModelForCausalLMIntegrationTest(unittest.TestCase):
1025
1025
"exaone" ,
1026
1026
"decilm" ,
1027
1027
"minicpm3" ,
1028
- "deepseek-v2" ,
1029
- "deepseek-v3" ,
1028
+ "deepseek" ,
1030
1029
)
1031
1030
1032
1031
@parameterized .expand (SUPPORTED_ARCHITECTURES )
Original file line number Diff line number Diff line change 51
51
"deberta" : "hf-internal-testing/tiny-random-deberta" ,
52
52
"deberta_v2" : "hf-internal-testing/tiny-random-DebertaV2Model" ,
53
53
"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" ,
56
55
"deit" : "hf-internal-testing/tiny-random-DeiTModel" ,
57
56
"convnext" : "hf-internal-testing/tiny-random-convnext" ,
58
57
"convnextv2" : "hf-internal-testing/tiny-random-ConvNextV2Model" ,
You can’t perform that action at this time.
0 commit comments