small fix

Querent-ai · May 17, 2024 · f33e70c · f33e70c
1 parent 6574791
commit f33e70c
Showing 1 changed file with 1 addition and 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
@@ -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()