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
20 changed files
--- a/tests/test_modeling_transfo_xl.py
+++ b/tests/test_modeling_transfo_xl.py
--- a/tests/test_modeling_xlm.py
+++ b/tests/test_modeling_xlm.py
--- a/tests/test_modeling_xlnet.py
+++ b/tests/test_modeling_xlnet.py
--- a/tests/test_optimization.py
+++ b/tests/test_optimization.py
--- a/tests/test_optimization_tf.py
+++ b/tests/test_optimization_tf.py
--- a/tests/test_pipelines.py
+++ b/tests/test_pipelines.py
--- a/tests/test_tokenization_albert.py
+++ b/tests/test_tokenization_albert.py
--- a/tests/test_tokenization_auto.py
+++ b/tests/test_tokenization_auto.py
--- a/tests/test_tokenization_bert.py
+++ b/tests/test_tokenization_bert.py
--- a/tests/test_tokenization_bert_japanese.py
+++ b/tests/test_tokenization_bert_japanese.py
--- a/tests/test_tokenization_common.py
+++ b/tests/test_tokenization_common.py
--- a/tests/test_tokenization_ctrl.py
+++ b/tests/test_tokenization_ctrl.py
--- a/tests/test_tokenization_distilbert.py
+++ b/tests/test_tokenization_distilbert.py
--- a/tests/test_tokenization_gpt2.py
+++ b/tests/test_tokenization_gpt2.py
--- a/tests/test_tokenization_openai.py
+++ b/tests/test_tokenization_openai.py
--- a/tests/test_tokenization_roberta.py
+++ b/tests/test_tokenization_roberta.py
--- a/tests/test_tokenization_t5.py
+++ b/tests/test_tokenization_t5.py
--- a/tests/test_tokenization_transfo_xl.py
+++ b/tests/test_tokenization_transfo_xl.py
--- a/tests/test_tokenization_utils.py
+++ b/tests/test_tokenization_utils.py
--- a/tests/test_tokenization_xlm.py
+++ b/tests/test_tokenization_xlm.py