-
Notifications
You must be signed in to change notification settings - Fork 2.7k
Haystack with Albert is awesome! XLNet question #23
Description
I am in the midst of evaluating Haystack with Albert and so far it looks awesome. Loving it, thanks for sharing.
I missed the whole Game of Thrones fantasy/drama phenomenon, so for a tutorial I could understand and relate-to, I went looking for other content to use with your Tutorial1_Basic_QA_Pipeline.ipynb notebook. Being a Porschephile I settled on:
import wikipedia
porsche_wikis = wikipedia.search("Porsche", results=25)
doc_dir = "data/porsche/"
for wiki in porsche_wikis:
html_page = wikipedia.page(title = wiki, auto_suggest = False)
text_file = open(doc_dir + wiki.replace('/', ' ') + ".txt", "w+")
text_file.write(html_page.content)
text_file.close()
print(wiki)
I can relate-to the above content and ask relevant questions of it "all day long". All other code in your notebook remains the same, except I use my Albert model for QA and it works well:
reader = FARMReader(model_name_or_path="ahotrod/albert_xxlargev1_squad2_512",
use_gpu=True)
For my application/project, I would like to also evaluate XLNet performance with Haystack but I am having trouble loading my XLNet model:
reader = FARMReader(model_name_or_path="ahotrod/xlnet_large_squad2_512",
use_gpu=True)
Attached is the complete terminal output text, but bottom-line the error I get is:
AttributeError: 'XLNetForQuestionAnswering' object has no attribute 'qa_outputs'
This XLNet model was fine-tuned on Transformers v2.1.1 and is the best I have because I and others are having problems fine-tuning XLNet_large under Transformers v2.4.1, huggingface/transformers#2651
Perhaps this fine-tuned XLNet model & Transformers v2.1.1 is not compatible/missing the attribute mentioned in the error message?
Looking forward to additional FARM/Haystack QA capabilities you have in the works, thanks for your efforts!