GPU text generation: mMoved the encoded_prompt to correct device
This commit is contained in:
alberduris
committed by
Lysandre Debut
Lysandre Debut
parent
dd4df80f0b
commit
81d6841b4b
0
tests/__init__.py
Executable file → Normal file
0
tests/__init__.py
Executable file → Normal file
0
tests/fixtures/empty.txt
vendored
Executable file → Normal file
0
tests/fixtures/empty.txt
vendored
Executable file → Normal file
0
tests/fixtures/input.txt
vendored
Executable file → Normal file
0
tests/fixtures/input.txt
vendored
Executable file → Normal file
0
tests/fixtures/sample_text.txt
vendored
Executable file → Normal file
0
tests/fixtures/sample_text.txt
vendored
Executable file → Normal file
0
tests/fixtures/spiece.model
vendored
Executable file → Normal file
0
tests/fixtures/spiece.model
vendored
Executable file → Normal file
0
tests/fixtures/test_sentencepiece.model
vendored
Executable file → Normal file
0
tests/fixtures/test_sentencepiece.model
vendored
Executable file → Normal file
0
tests/test_configuration_common.py
Executable file → Normal file
0
tests/test_configuration_common.py
Executable file → Normal file
0
tests/test_hf_api.py
Executable file → Normal file
0
tests/test_hf_api.py
Executable file → Normal file
0
tests/test_model_card.py
Executable file → Normal file
0
tests/test_model_card.py
Executable file → Normal file
0
tests/test_modeling_albert.py
Executable file → Normal file
0
tests/test_modeling_albert.py
Executable file → Normal file
0
tests/test_modeling_auto.py
Executable file → Normal file
0
tests/test_modeling_auto.py
Executable file → Normal file
0
tests/test_modeling_bert.py
Executable file → Normal file
0
tests/test_modeling_bert.py
Executable file → Normal file
0
tests/test_modeling_common.py
Executable file → Normal file
0
tests/test_modeling_common.py
Executable file → Normal file
0
tests/test_modeling_ctrl.py
Executable file → Normal file
0
tests/test_modeling_ctrl.py
Executable file → Normal file
0
tests/test_modeling_distilbert.py
Executable file → Normal file
0
tests/test_modeling_distilbert.py
Executable file → Normal file
0
tests/test_modeling_encoder_decoder.py
Executable file → Normal file
0
tests/test_modeling_encoder_decoder.py
Executable file → Normal file
0
tests/test_modeling_gpt2.py
Executable file → Normal file
0
tests/test_modeling_gpt2.py
Executable file → Normal file
0
tests/test_modeling_openai.py
Executable file → Normal file
0
tests/test_modeling_openai.py
Executable file → Normal file
0
tests/test_modeling_roberta.py
Executable file → Normal file
0
tests/test_modeling_roberta.py
Executable file → Normal file
0
tests/test_modeling_t5.py
Executable file → Normal file
0
tests/test_modeling_t5.py
Executable file → Normal file
0
tests/test_modeling_tf_albert.py
Executable file → Normal file
0
tests/test_modeling_tf_albert.py
Executable file → Normal file
0
tests/test_modeling_tf_auto.py
Executable file → Normal file
0
tests/test_modeling_tf_auto.py
Executable file → Normal file
0
tests/test_modeling_tf_bert.py
Executable file → Normal file
0
tests/test_modeling_tf_bert.py
Executable file → Normal file
0
tests/test_modeling_tf_common.py
Executable file → Normal file
0
tests/test_modeling_tf_common.py
Executable file → Normal file
0
tests/test_modeling_tf_ctrl.py
Executable file → Normal file
0
tests/test_modeling_tf_ctrl.py
Executable file → Normal file
0
tests/test_modeling_tf_distilbert.py
Executable file → Normal file
0
tests/test_modeling_tf_distilbert.py
Executable file → Normal file
0
tests/test_modeling_tf_gpt2.py
Executable file → Normal file
0
tests/test_modeling_tf_gpt2.py
Executable file → Normal file
0
tests/test_modeling_tf_openai_gpt.py
Executable file → Normal file
0
tests/test_modeling_tf_openai_gpt.py
Executable file → Normal file
0
tests/test_modeling_tf_roberta.py
Executable file → Normal file
0
tests/test_modeling_tf_roberta.py
Executable file → Normal file
0
tests/test_modeling_tf_t5.py
Executable file → Normal file
0
tests/test_modeling_tf_t5.py
Executable file → Normal file
0
tests/test_modeling_tf_transfo_xl.py
Executable file → Normal file
0
tests/test_modeling_tf_transfo_xl.py
Executable file → Normal file
0
tests/test_modeling_tf_xlm.py
Executable file → Normal file
0
tests/test_modeling_tf_xlm.py
Executable file → Normal file
0
tests/test_modeling_tf_xlnet.py
Executable file → Normal file
0
tests/test_modeling_tf_xlnet.py
Executable file → Normal file
0
tests/test_modeling_transfo_xl.py
Executable file → Normal file
0
tests/test_modeling_transfo_xl.py
Executable file → Normal file
0
tests/test_modeling_xlm.py
Executable file → Normal file
0
tests/test_modeling_xlm.py
Executable file → Normal file
0
tests/test_modeling_xlnet.py
Executable file → Normal file
0
tests/test_modeling_xlnet.py
Executable file → Normal file
0
tests/test_optimization.py
Executable file → Normal file
0
tests/test_optimization.py
Executable file → Normal file
0
tests/test_optimization_tf.py
Executable file → Normal file
0
tests/test_optimization_tf.py
Executable file → Normal file
0
tests/test_pipelines.py
Executable file → Normal file
0
tests/test_pipelines.py
Executable file → Normal file
0
tests/test_tokenization_albert.py
Executable file → Normal file
0
tests/test_tokenization_albert.py
Executable file → Normal file
0
tests/test_tokenization_auto.py
Executable file → Normal file
0
tests/test_tokenization_auto.py
Executable file → Normal file
0
tests/test_tokenization_bert.py
Executable file → Normal file
0
tests/test_tokenization_bert.py
Executable file → Normal file
0
tests/test_tokenization_bert_japanese.py
Executable file → Normal file
0
tests/test_tokenization_bert_japanese.py
Executable file → Normal file
0
tests/test_tokenization_common.py
Executable file → Normal file
0
tests/test_tokenization_common.py
Executable file → Normal file
0
tests/test_tokenization_ctrl.py
Executable file → Normal file
0
tests/test_tokenization_ctrl.py
Executable file → Normal file
0
tests/test_tokenization_distilbert.py
Executable file → Normal file
0
tests/test_tokenization_distilbert.py
Executable file → Normal file
0
tests/test_tokenization_gpt2.py
Executable file → Normal file
0
tests/test_tokenization_gpt2.py
Executable file → Normal file
0
tests/test_tokenization_openai.py
Executable file → Normal file
0
tests/test_tokenization_openai.py
Executable file → Normal file
0
tests/test_tokenization_roberta.py
Executable file → Normal file
0
tests/test_tokenization_roberta.py
Executable file → Normal file
0
tests/test_tokenization_t5.py
Executable file → Normal file
0
tests/test_tokenization_t5.py
Executable file → Normal file
0
tests/test_tokenization_transfo_xl.py
Executable file → Normal file
0
tests/test_tokenization_transfo_xl.py
Executable file → Normal file
0
tests/test_tokenization_utils.py
Executable file → Normal file
0
tests/test_tokenization_utils.py
Executable file → Normal file
0
tests/test_tokenization_xlm.py
Executable file → Normal file
0
tests/test_tokenization_xlm.py
Executable file → Normal file
0
tests/test_tokenization_xlnet.py
Executable file → Normal file
0
tests/test_tokenization_xlnet.py
Executable file → Normal file
0
tests/utils.py
Executable file → Normal file
0
tests/utils.py
Executable file → Normal file
Reference in New Issue
Block a user