tests/test_modeling_rag.py · 43f46aa7fdd75e51b4a884a6c0f00b5d9a0aae8b · chenpangpang / transformers · GitLab

Find file Blame History Permalink

[RAG] Fix rag from pretrained question encoder generator behavior (#11962) · 43f46aa7
Patrick von Platen authored Jun 02, 2021
```
* fix_torch_device_generate_test

* remove @

* fix rag from pretrained loading

* add test

* uplaod

* finish
```
43f46aa7

test_modeling_rag.py 43.8 KB