yujunjun/haystack
1
0
Fork 0
mirror of https://github.com/deepset-ai/haystack.git synced 2025-12-30 00:30:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

Remove NER and text classification from model conversion (#1536)

Browse Source
This commit is contained in:
Julian Risch 2021-09-29 13:35:59 +02:00 committed by GitHub
parent 0e7338f0c6
commit 9ed726923c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
haystack/modeling/conversion/transformers.py
View File

@ -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
return transformers_model