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