GPU text generation: mMoved the encoded_prompt to correct device
This commit is contained in:
committed by
Lysandre Debut
parent
dd4df80f0b
commit
81d6841b4b
0
docs/source/model_doc/albert.rst
Executable file → Normal file
0
docs/source/model_doc/albert.rst
Executable file → Normal file
0
docs/source/model_doc/auto.rst
Executable file → Normal file
0
docs/source/model_doc/auto.rst
Executable file → Normal file
0
docs/source/model_doc/bert.rst
Executable file → Normal file
0
docs/source/model_doc/bert.rst
Executable file → Normal file
0
docs/source/model_doc/camembert.rst
Executable file → Normal file
0
docs/source/model_doc/camembert.rst
Executable file → Normal file
0
docs/source/model_doc/ctrl.rst
Executable file → Normal file
0
docs/source/model_doc/ctrl.rst
Executable file → Normal file
0
docs/source/model_doc/distilbert.rst
Executable file → Normal file
0
docs/source/model_doc/distilbert.rst
Executable file → Normal file
0
docs/source/model_doc/gpt.rst
Executable file → Normal file
0
docs/source/model_doc/gpt.rst
Executable file → Normal file
0
docs/source/model_doc/gpt2.rst
Executable file → Normal file
0
docs/source/model_doc/gpt2.rst
Executable file → Normal file
0
docs/source/model_doc/roberta.rst
Executable file → Normal file
0
docs/source/model_doc/roberta.rst
Executable file → Normal file
0
docs/source/model_doc/transformerxl.rst
Executable file → Normal file
0
docs/source/model_doc/transformerxl.rst
Executable file → Normal file
0
docs/source/model_doc/xlm.rst
Executable file → Normal file
0
docs/source/model_doc/xlm.rst
Executable file → Normal file
0
docs/source/model_doc/xlnet.rst
Executable file → Normal file
0
docs/source/model_doc/xlnet.rst
Executable file → Normal file
Reference in New Issue
Block a user