add mistral config

explosion · Jan 22, 2024 · 4f30184 · 4f30184
1 parent c8914dc
commit 4f30184
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/tutorials/llm_clinical_trials/configs/ner_mistral.cfg b/tutorials/llm_clinical_trials/configs/ner_mistral.cfg
@@ -0,0 +1,17 @@
+[nlp]
+lang = "en"
+pipeline = ["llm"]
+batch_size = 128
+
+[components]
+
+[components.llm]
+factory = "llm"
+
+[components.llm.model]
+@llm_models = "spacy.Mistral.v1"
+name = "Mistral-7B-v0.1"
+
+[components.llm.task]
+@llm_tasks = "spacy.NER.v2"
+labels = ["PERSON", "LOCATION"]