yujunjun/haystack
1
0
Fork 0
mirror of https://github.com/deepset-ai/haystack.git synced 2026-01-06 12:07:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

Bump to FARM 0.8.0, torch 1.8.1 and transformers 4.6.1 (#1192)

Browse Source 
* bump to FARM 0.8.0, which in turn bumps torch 1.8.1 and transformers 4.6.1 (#1192)

* Replace deprecated force_bos_token_to_be_generated parameter
This commit is contained in:
Julian Risch 2021-06-14 13:00:41 +02:00 committed by GitHub
parent f583d0bfaf
commit 1c31589b43
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
haystack/graph_retriever/text_to_sparql.py
View File

@ -28,7 +28,7 @@ class Text2SparqlRetriever(BaseGraphRetriever):
self.knowledge_graph = knowledge_graph
# TODO We should extend this to any seq2seq models and use the AutoModel class
self.model = BartForConditionalGeneration.from_pretrained(model_name_or_path, force_bos_token_to_be_generated=True)
self.model = BartForConditionalGeneration.from_pretrained(model_name_or_path, forced_bos_token_id=0)
self.tok = BartTokenizer.from_pretrained(model_name_or_path)
self.top_k = top_k

4
requirements.txt
View File

@ -1,4 +1,4 @@
farm==0.7.1
farm==0.8.0
--find-links=https://download.pytorch.org/whl/torch_stable.html
fastapi
uvicorn
@ -31,4 +31,4 @@ pymilvus
#webdriver-manager
SPARQLWrapper
mmh3
weaviate-client
weaviate-client