From 9ed726923c66316fdd39590753f2ccc8ec1f9bfa Mon Sep 17 00:00:00 2001 From: Julian Risch Date: Wed, 29 Sep 2021 13:35:59 +0200 Subject: [PATCH] Remove NER and text classification from model conversion (#1536) --- haystack/modeling/conversion/transformers.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/haystack/modeling/conversion/transformers.py b/haystack/modeling/conversion/transformers.py index 701f5ceb7..541a9bdb7 100644 --- a/haystack/modeling/conversion/transformers.py +++ b/haystack/modeling/conversion/transformers.py @@ -72,7 +72,7 @@ class Converter: task_type = "question_answering" else: logger.error("Could not infer task type from model config. Please provide task type manually. " - "('lm', 'question_answering', 'regression', 'text_classification', 'ner' or 'embeddings')") + "('question_answering' or 'embeddings')") if task_type == "question_answering": @@ -101,4 +101,4 @@ class Converter: transformers_model.qa_outputs.load_state_dict( prediction_head.feed_forward.feed_forward[0].state_dict()) - return transformers_model \ No newline at end of file + return transformers_model