add more features to phi-2 model

openvinotoolkit · Apr 2, 2024 · 8e987c4 · 8e987c4
1 parent e15058d
commit 8e987c4
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 7 deletions.
diff --git a/modules/openvino_code/package-lock.json b/modules/openvino_code/package-lock.json
diff --git a/modules/openvino_code/package.json b/modules/openvino_code/package.json
@@ -1,7 +1,7 @@
 {
   "publisher": "OpenVINO",
   "name": "openvino-code-completion",
-  "version": "0.0.13",
+  "version": "0.0.14",
   "displayName": "OpenVINO Code Completion",
   "description": "VSCode extension for AI code completion with OpenVINO",
   "icon": "media/logo.png",

diff --git a/modules/openvino_code/shared/features.ts b/modules/openvino_code/shared/features.ts
@@ -2,4 +2,6 @@ export enum Features {
   CODE_COMPLETION = 'Code Completion',
   SUMMARIZATION = 'Summarization',
   FIM = 'Fill-in-the-middle',
+  QA_FORMAT = 'QA Format',
+  CHAT_FORMAT = 'Chat Format',
 }
diff --git a/modules/openvino_code/shared/model.ts b/modules/openvino_code/shared/model.ts
@@ -29,5 +29,5 @@ export const MODEL_SUPPORTED_FEATURES: Record<ModelName, Features[]> = {
   [ModelName.DECICODER_1B_OPENVINO_INT8]: [Features.CODE_COMPLETION, Features.SUMMARIZATION],
   [ModelName.STABLECODE_COMPLETION_ALPHA_3B_4K_OPENVINO_INT8]: [Features.CODE_COMPLETION, Features.SUMMARIZATION],
   [ModelName.DEEPSEEK_CODER_1_3B]: [Features.CODE_COMPLETION, Features.SUMMARIZATION, Features.FIM],
-  [ModelName.PHI_2_2_7B]: [Features.CODE_COMPLETION],
+  [ModelName.PHI_2_2_7B]: [Features.CODE_COMPLETION, Features.SUMMARIZATION, Features.QA_FORMAT, Features.CHAT_FORMAT],
 };