From de1b15ee286daa2593810b09243b000e061ab35a Mon Sep 17 00:00:00 2001 From: Kumari Date: Wed, 1 Nov 2023 00:17:17 -0700 Subject: [PATCH] Adding stablecode-3m model to Open Vino code completion extension --- modules/openvino_code/package.json | 3 ++- modules/openvino_code/shared/model.ts | 5 +++++ 2 files changed, 7 insertions(+), 1 deletion(-) diff --git a/modules/openvino_code/package.json b/modules/openvino_code/package.json index 93476bff9..0552ff9b8 100644 --- a/modules/openvino_code/package.json +++ b/modules/openvino_code/package.json @@ -171,7 +171,8 @@ "default": "codet5p-220m-py", "enum": [ "codet5p-220m-py", - "decicoder-1b-openvino-int8" + "decicoder-1b-openvino-int8", + "stablecode-completion-3b" ], "description": "Which model to use for code generation." }, diff --git a/modules/openvino_code/shared/model.ts b/modules/openvino_code/shared/model.ts index 660cb383b..0fae6467a 100644 --- a/modules/openvino_code/shared/model.ts +++ b/modules/openvino_code/shared/model.ts @@ -3,19 +3,24 @@ import { Features } from './features'; enum ModelId { CODE_T5_220M = 'Salesforce/codet5p-220m-py', DECICODER_1B_OPENVINO_INT8 = 'chgk13/decicoder-1b-openvino-int8', + STABLECODE_COMPLETION_ALPHA_3B_4K_OPENVINO_INT8 = 'chgk13/stablecode-completion-alpha-3b-4k-openvino-int8', } export enum ModelName { CODE_T5_220M = 'codet5p-220m-py', DECICODER_1B_OPENVINO_INT8 = 'decicoder-1b-openvino-int8', + STABLECODE_COMPLETION_ALPHA_3B_4K_OPENVINO_INT8 = 'stablecode-completion-3b', + } export const MODEL_NAME_TO_ID_MAP: Record = { [ModelName.CODE_T5_220M]: ModelId.CODE_T5_220M, [ModelName.DECICODER_1B_OPENVINO_INT8]: ModelId.DECICODER_1B_OPENVINO_INT8, + [ModelName.STABLECODE_COMPLETION_ALPHA_3B_4K_OPENVINO_INT8]: ModelId.STABLECODE_COMPLETION_ALPHA_3B_4K_OPENVINO_INT8, }; export const MODEL_SUPPORTED_FEATURES: Record = { [ModelName.CODE_T5_220M]: [Features.CODE_COMPLETION], [ModelName.DECICODER_1B_OPENVINO_INT8]: [Features.CODE_COMPLETION, Features.SUMMARIZATION], + [ModelName.STABLECODE_COMPLETION_ALPHA_3B_4K_OPENVINO_INT8]: [Features.CODE_COMPLETION], };