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/templates/adding_a_new_model/configuration_xxx.py
+++ b/templates/adding_a_new_model/configuration_xxx.py
--- a/templates/adding_a_new_model/modeling_tf_xxx.py
+++ b/templates/adding_a_new_model/modeling_tf_xxx.py
--- a/templates/adding_a_new_model/modeling_xxx.py
+++ b/templates/adding_a_new_model/modeling_xxx.py
--- a/templates/adding_a_new_model/tests/test_modeling_tf_xxx.py
+++ b/templates/adding_a_new_model/tests/test_modeling_tf_xxx.py
--- a/templates/adding_a_new_model/tests/test_modeling_xxx.py
+++ b/templates/adding_a_new_model/tests/test_modeling_xxx.py
--- a/templates/adding_a_new_model/tests/test_tokenization_xxx.py
+++ b/templates/adding_a_new_model/tests/test_tokenization_xxx.py
--- a/templates/adding_a_new_model/tokenization_xxx.py
+++ b/templates/adding_a_new_model/tokenization_xxx.py
--- a/tests/__init__.py
+++ b/tests/__init__.py
--- a/tests/fixtures/empty.txt
+++ b/tests/fixtures/empty.txt
--- a/tests/fixtures/input.txt
+++ b/tests/fixtures/input.txt
--- a/tests/fixtures/sample_text.txt
+++ b/tests/fixtures/sample_text.txt
--- a/tests/fixtures/spiece.model
+++ b/tests/fixtures/spiece.model
--- a/tests/fixtures/test_sentencepiece.model
+++ b/tests/fixtures/test_sentencepiece.model
--- a/tests/test_configuration_common.py
+++ b/tests/test_configuration_common.py
--- a/tests/test_hf_api.py
+++ b/tests/test_hf_api.py
--- a/tests/test_model_card.py
+++ b/tests/test_model_card.py
--- a/tests/test_modeling_albert.py
+++ b/tests/test_modeling_albert.py
--- a/tests/test_modeling_auto.py
+++ b/tests/test_modeling_auto.py
--- a/tests/test_modeling_bert.py
+++ b/tests/test_modeling_bert.py
--- a/tests/test_modeling_common.py
+++ b/tests/test_modeling_common.py