@@ -5,25 +5,29 @@ enum ModelId {
5
5
DECICODER_1B_OPENVINO_INT8 = 'chgk13/decicoder-1b-openvino-int8' ,
6
6
STABLECODE_COMPLETION_ALPHA_3B_4K_OPENVINO_INT8 = 'chgk13/stablecode-completion-alpha-3b-4k-openvino-int8' ,
7
7
DEEPSEEK_CODER_1_3B = 'kumarijy/deepseek-code-1.3b_base_ov_int8' ,
8
+ PHI_2_2_7B = 'kumarijy/phi-2-2.7b_ov_int8' ,
8
9
}
9
10
10
11
export enum ModelName {
11
12
CODE_T5_220M = 'code-t5' ,
12
13
DECICODER_1B_OPENVINO_INT8 = 'decicoder-1b-openvino' ,
13
14
STABLECODE_COMPLETION_ALPHA_3B_4K_OPENVINO_INT8 = 'stablecode-completion' ,
14
15
DEEPSEEK_CODER_1_3B = 'deepseek-coder' ,
16
+ PHI_2_2_7B = 'phi-2' ,
15
17
}
16
18
17
19
export const MODEL_NAME_TO_ID_MAP : Record < ModelName , ModelId > = {
18
20
[ ModelName . CODE_T5_220M ] : ModelId . CODE_T5_220M ,
19
21
[ ModelName . DECICODER_1B_OPENVINO_INT8 ] : ModelId . DECICODER_1B_OPENVINO_INT8 ,
20
22
[ ModelName . STABLECODE_COMPLETION_ALPHA_3B_4K_OPENVINO_INT8 ] : ModelId . STABLECODE_COMPLETION_ALPHA_3B_4K_OPENVINO_INT8 ,
21
23
[ ModelName . DEEPSEEK_CODER_1_3B ] : ModelId . DEEPSEEK_CODER_1_3B ,
24
+ [ ModelName . PHI_2_2_7B ] : ModelId . PHI_2_2_7B ,
22
25
} ;
23
26
24
27
export const MODEL_SUPPORTED_FEATURES : Record < ModelName , Features [ ] > = {
25
28
[ ModelName . CODE_T5_220M ] : [ Features . CODE_COMPLETION ] ,
26
29
[ ModelName . DECICODER_1B_OPENVINO_INT8 ] : [ Features . CODE_COMPLETION , Features . SUMMARIZATION ] ,
27
30
[ ModelName . STABLECODE_COMPLETION_ALPHA_3B_4K_OPENVINO_INT8 ] : [ Features . CODE_COMPLETION , Features . SUMMARIZATION ] ,
28
31
[ ModelName . DEEPSEEK_CODER_1_3B ] : [ Features . CODE_COMPLETION , Features . SUMMARIZATION , Features . FIM ] ,
32
+ [ ModelName . PHI_2_2_7B ] : [ Features . CODE_COMPLETION ] ,
29
33
} ;
0 commit comments