GPU text generation: mMoved the encoded_prompt to correct device
This commit is contained in:
committed by
Lysandre Debut
parent
dd4df80f0b
commit
81d6841b4b
0
notebooks/Comparing-PT-and-TF-models.ipynb
Executable file → Normal file
0
notebooks/Comparing-PT-and-TF-models.ipynb
Executable file → Normal file
0
notebooks/Comparing-TF-and-PT-models-MLM-NSP.ipynb
Executable file → Normal file
0
notebooks/Comparing-TF-and-PT-models-MLM-NSP.ipynb
Executable file → Normal file
0
notebooks/Comparing-TF-and-PT-models-SQuAD.ipynb
Executable file → Normal file
0
notebooks/Comparing-TF-and-PT-models-SQuAD.ipynb
Executable file → Normal file
0
notebooks/Comparing-TF-and-PT-models.ipynb
Executable file → Normal file
0
notebooks/Comparing-TF-and-PT-models.ipynb
Executable file → Normal file
Reference in New Issue
Block a user