Adding stablecode-3m model to Open Vino code completion extension

openvinotoolkit · Nov 1, 2023 · de1b15e · de1b15e
1 parent de1c843
commit de1b15e
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git 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
@@ -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, ModelId> = {
   [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, Features[]> = {
   [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],
 };