From f33e70ccbff965cc5fa04f542d6ca1adb5e95505 Mon Sep 17 00:00:00 2001 From: ngupta10 Date: Fri, 17 May 2024 15:39:57 +0530 Subject: [PATCH] small fix --- .../transformers/gpt_llm_bert_ner_or_fixed_entities_set_ner.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/querent/core/transformers/gpt_llm_bert_ner_or_fixed_entities_set_ner.py b/querent/core/transformers/gpt_llm_bert_ner_or_fixed_entities_set_ner.py index 026ff8f0..a87429ed 100644 --- a/querent/core/transformers/gpt_llm_bert_ner_or_fixed_entities_set_ner.py +++ b/querent/core/transformers/gpt_llm_bert_ner_or_fixed_entities_set_ner.py @@ -73,7 +73,7 @@ def _initialize_config(self, config): self.user_context = config.user_context def _initialize_models(self, config): - self.rel_model_initialized = self.model_manager.get_model(config.rel_model_path) + self.rel_model_initialized = self.model_manager.get_model(config.rel_model_name) self.nlp_model = NER_LLM.set_nlp_model(config.spacy_model_path) self.nlp_model = NER_LLM.get_class_variable() self.create_emb = EmbeddingStore()