Skip to content

Commit eaaff11

Browse files
committed
updated model selection
1 parent 7b57e4c commit eaaff11

File tree

1 file changed

+9
-3
lines changed

1 file changed

+9
-3
lines changed

frontend/src/components/Classifier/Dialog/steps/TrainingSettingsStep.tsx

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -50,8 +50,15 @@ const transformerModelOptions: FreeSoloOptions[] = [
5050

5151
const embeddingModelOptions: FreeSoloOptions[] = [
5252
{ value: "Alibaba-NLP/gte-modernbert-base", label: "gte-modernbert-base (EN)" },
53-
{ value: "jinaai/jina-embeddings-v3", label: "jina-embeddings-v3 (MULTI)" },
54-
{ value: "Qwen/Qwen3-Embedding-0.6B", label: "Qwen3-Embedding-0.6B (MULTI)" },
53+
{ value: "intfloat/multilingual-e5-small", label: "multilingual-e5-small (MULTI)" },
54+
{ value: "intfloat/multilingual-e5-large", label: "multilingual-e5-large (MULTI)" },
55+
{
56+
value: "sentence-transformers/paraphrase-multilingual-mpnet-base-v2",
57+
label: "paraphrase-multilingual-mpnet-base-v2 (MULTI)",
58+
},
59+
// { value: "google/embeddinggemma-300m", label: "embeddinggemma-300m (MULTI)" }, TODO: update hf library
60+
// { value: "jinaai/jina-embeddings-v3", label: "jina-embeddings-v3 (MULTI)" }, TODO: update st library
61+
// { value: "Qwen/Qwen3-Embedding-0.6B", label: "Qwen3-Embedding-0.6B (MULTI)" }, TODO: update hf library
5562
];
5663

5764
const adapterOptions = ["No Adapter", "LoRA", "LoHa", "AdaLoRA", "RandLora"];
@@ -227,7 +234,6 @@ function TrainingSettingsStep() {
227234
<FormSwitch
228235
name="earlyStopping"
229236
control={control}
230-
rules={{ required: "Required" }}
231237
switchProps={{ size: "medium", color: "primary" }}
232238
boxProps={{ sx: { ml: 2 } }}
233239
/>

0 commit comments

Comments
 (0)