Skip to content
Snippets Groups Projects
Commit 712fde4b authored by Abhisek Mazumdar's avatar Abhisek Mazumdar Committed by Eleo Basili
Browse files

Issue #3402710: Update NLP Cloud Keywords and keyphrases extraction models

parent 06732c8d
No related branches found
No related tags found
Loading
Pipeline #72833 canceled
......@@ -43,7 +43,7 @@ class NLPCloudKeywordsKeyphrasesExtraction extends NPLCloudBase {
'#type' => 'select',
'#title' => $this->t('Model'),
'#options' => $this->getSupportedModels(),
'#default_value' => $this->configuration['model'] ?? 'fast-gpt-j',
'#default_value' => $this->configuration['model'] ?? 'finetuned-llama-2-70b',
'#description' => $this->t("Specifies the model which you want to use for keywords extraction."),
];
return $form;
......@@ -93,8 +93,8 @@ class NLPCloudKeywordsKeyphrasesExtraction extends NPLCloudBase {
*/
private function getSupportedModels(): array {
return [
'fast-gpt-j' => $this->t('Fast GPT-J'),
'finetuned-gpt-neox-20b' => $this->t('Finetuned GPT-NeoX 20B'),
'finetuned-llama-2-70b' => $this->t('Fine-tuned LLaMA 2 70B'),
'dolphin' => $this->t('Dolphin'),
];
}
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment