GPU text generation: mMoved the encoded_prompt to correct device
This commit is contained in:
committed by
Lysandre Debut
parent
dd4df80f0b
commit
81d6841b4b
0
examples/summarization/README.md
Executable file → Normal file
0
examples/summarization/README.md
Executable file → Normal file
0
examples/summarization/configuration_bertabs.py
Executable file → Normal file
0
examples/summarization/configuration_bertabs.py
Executable file → Normal file
0
examples/summarization/convert_bertabs_original_pytorch_checkpoint.py
Executable file → Normal file
0
examples/summarization/convert_bertabs_original_pytorch_checkpoint.py
Executable file → Normal file
0
examples/summarization/modeling_bertabs.py
Executable file → Normal file
0
examples/summarization/modeling_bertabs.py
Executable file → Normal file
0
examples/summarization/requirements.txt
Executable file → Normal file
0
examples/summarization/requirements.txt
Executable file → Normal file
0
examples/summarization/run_summarization.py
Executable file → Normal file
0
examples/summarization/run_summarization.py
Executable file → Normal file
0
examples/summarization/test_utils_summarization.py
Executable file → Normal file
0
examples/summarization/test_utils_summarization.py
Executable file → Normal file
0
examples/summarization/utils_summarization.py
Executable file → Normal file
0
examples/summarization/utils_summarization.py
Executable file → Normal file
Reference in New Issue
Block a user