GPU text generation: mMoved the encoded_prompt to correct device
This commit is contained in:
committed by
Lysandre Debut
parent
dd4df80f0b
commit
81d6841b4b
0
src/transformers/data/__init__.py
Executable file → Normal file
0
src/transformers/data/__init__.py
Executable file → Normal file
0
src/transformers/data/metrics/__init__.py
Executable file → Normal file
0
src/transformers/data/metrics/__init__.py
Executable file → Normal file
0
src/transformers/data/metrics/squad_metrics.py
Executable file → Normal file
0
src/transformers/data/metrics/squad_metrics.py
Executable file → Normal file
0
src/transformers/data/processors/__init__.py
Executable file → Normal file
0
src/transformers/data/processors/__init__.py
Executable file → Normal file
0
src/transformers/data/processors/glue.py
Executable file → Normal file
0
src/transformers/data/processors/glue.py
Executable file → Normal file
0
src/transformers/data/processors/squad.py
Executable file → Normal file
0
src/transformers/data/processors/squad.py
Executable file → Normal file
0
src/transformers/data/processors/utils.py
Executable file → Normal file
0
src/transformers/data/processors/utils.py
Executable file → Normal file
0
src/transformers/data/processors/xnli.py
Executable file → Normal file
0
src/transformers/data/processors/xnli.py
Executable file → Normal file
Reference in New Issue
Block a user