GPU text generation: mMoved the encoded_prompt to correct device

This commit is contained in:
alberduris
2019-12-31 13:24:54 +01:00
committed by Lysandre Debut
parent dd4df80f0b
commit 81d6841b4b
285 changed files with 0 additions and 0 deletions

0
src/transformers/data/__init__.py Executable file → Normal file
View File

0
src/transformers/data/metrics/__init__.py Executable file → Normal file
View File

0
src/transformers/data/metrics/squad_metrics.py Executable file → Normal file
View File

0
src/transformers/data/processors/__init__.py Executable file → Normal file
View File

0
src/transformers/data/processors/glue.py Executable file → Normal file
View File

0
src/transformers/data/processors/squad.py Executable file → Normal file
View File

0
src/transformers/data/processors/utils.py Executable file → Normal file
View File

0
src/transformers/data/processors/xnli.py Executable file → Normal file
View File