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 @@ -1000,7 +1000,7 @@ class OVModelForCausalLMIntegrationTest(unittest.TestCase):
1000
1000
1001
1001
# openvino 2025.0 required for disabling check_trace
1002
1002
if is_openvino_version (">=" , "2025.0" ):
1003
- SUPPORTED_ARCHITECTURES += ( "deepseek-v2" , "deepseek-v3" )
1003
+ SUPPORTED_ARCHITECTURES += "deepseek"
1004
1004
1005
1005
# gptq and awq install disabled for windows test environment
1006
1006
if platform .system () != "Windows" :
@@ -1030,8 +1030,7 @@ class OVModelForCausalLMIntegrationTest(unittest.TestCase):
1030
1030
"exaone" ,
1031
1031
"decilm" ,
1032
1032
"minicpm3" ,
1033
- "deepseek-v2" ,
1034
- "deepseek-v3" ,
1033
+ "deepseek" ,
1035
1034
)
1036
1035
1037
1036
@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