[VSCode Extension] Adding stablecode-3b-int8 model #753
Conversation
modules/openvino_code/package.json
Outdated
| @@ -171,7 +171,8 @@ | |||
| "default": "codet5p-220m-py", | |||
| "enum": [ | |||
| "codet5p-220m-py", | |||
| "decicoder-1b-openvino-int8" | |||
| "decicoder-1b-openvino-int8", | |||
| "stablecode-completion-3b" | |||
There was a problem hiding this comment.
Please add int8 to the model name: "stablecode-completion-3b-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], |
There was a problem hiding this comment.
The model might support the summarization feature too, you need to check it.
There was a problem hiding this comment.
Please increment extension patch version in package.json file (L4, "version" field) from 0.0.5 to 0.0.6
| } | ||
|
|
||
| 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', |
There was a problem hiding this comment.
Please add -int8 here too.
| } | ||
|
|
||
| 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', |
There was a problem hiding this comment.
New model name should also be added to side panel select options: side-panel-ui/src/components/sections/ServerSection/ModelSelect/ModelSelect.tsx (const options: SelectOptionProps<ModelName>[], L6-L9)
There was a problem hiding this comment.
Also, I would suggest to make it more dynamic, e.g.
const options: SelectOptionProps<ModelName>[] = Object.values(ModelName).map((value) => ({ value }));
yatarkan
changed the title
|
It seems like the old |
apaniukov
changed the title
made changes to the relevant files in order to add stablecode-3m model to OV code completion extension @apaniukov