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