Pavel Soriano
commited on
Commit
·
a0bc241
1
Parent(s):
647da32
updated Readme
Browse files
README.md
CHANGED
|
@@ -109,17 +109,17 @@ embeddings = model.forward(input_ids).pooler_output
|
|
| 109 |
print(embeddings)
|
| 110 |
```
|
| 111 |
|
| 112 |
-
And with `haystack
|
| 113 |
```
|
| 114 |
-
retriever = DensePassageRetriever(
|
| 115 |
-
|
| 116 |
-
|
| 117 |
-
|
| 118 |
-
|
| 119 |
-
|
| 120 |
-
|
| 121 |
-
)
|
| 122 |
```
|
|
|
|
| 123 |
## Acknowledgments
|
| 124 |
|
| 125 |
This work was performed using HPC resources from GENCI–IDRIS (Grant 2020-AD011011224).
|
|
|
|
| 109 |
print(embeddings)
|
| 110 |
```
|
| 111 |
|
| 112 |
+
And with `haystack`, we use it as a retriever:
|
| 113 |
```
|
| 114 |
+
retriever = DensePassageRetriever(
|
| 115 |
+
document_store=document_store,
|
| 116 |
+
query_embedding_model="etalab-ia/dpr-question_encoder-fr_qa-camembert",
|
| 117 |
+
passage_embedding_model="etalab-ia/dpr-ctx_encoder-fr_qa-camembert",
|
| 118 |
+
model_version=dpr_model_tag,
|
| 119 |
+
infer_tokenizer_classes=True,
|
| 120 |
+
)
|
|
|
|
| 121 |
```
|
| 122 |
+
|
| 123 |
## Acknowledgments
|
| 124 |
|
| 125 |
This work was performed using HPC resources from GENCI–IDRIS (Grant 2020-AD011011224).
|