GPU text generation: mMoved the encoded_prompt to correct device

81d6841b · alberduris · Lysandre Debut · dd4df80f · 81d6841b · 81d6841b
Commit 81d6841b authored Dec 31, 2019 by alberduris Committed by Lysandre Debut Jan 06, 2020
5 changed files
--- a/tests/test_tokenization_xlnet.py
+++ b/tests/test_tokenization_xlnet.py
--- a/tests/utils.py
+++ b/tests/utils.py
--- a/utils/download_glue_data.py
+++ b/utils/download_glue_data.py
--- a/utils/link_tester.py
+++ b/utils/link_tester.py
--- a/valohai.yaml
+++ b/valohai.yaml