GPU text generation: mMoved the encoded_prompt to correct device
This commit is contained in:
committed by
Lysandre Debut
parent
dd4df80f0b
commit
81d6841b4b
0
templates/adding_a_new_model/README.md
Executable file → Normal file
0
templates/adding_a_new_model/README.md
Executable file → Normal file
0
templates/adding_a_new_model/configuration_xxx.py
Executable file → Normal file
0
templates/adding_a_new_model/configuration_xxx.py
Executable file → Normal file
0
templates/adding_a_new_model/modeling_tf_xxx.py
Executable file → Normal file
0
templates/adding_a_new_model/modeling_tf_xxx.py
Executable file → Normal file
0
templates/adding_a_new_model/modeling_xxx.py
Executable file → Normal file
0
templates/adding_a_new_model/modeling_xxx.py
Executable file → Normal file
0
templates/adding_a_new_model/tests/test_modeling_tf_xxx.py
Executable file → Normal file
0
templates/adding_a_new_model/tests/test_modeling_tf_xxx.py
Executable file → Normal file
0
templates/adding_a_new_model/tests/test_modeling_xxx.py
Executable file → Normal file
0
templates/adding_a_new_model/tests/test_modeling_xxx.py
Executable file → Normal file
0
templates/adding_a_new_model/tests/test_tokenization_xxx.py
Executable file → Normal file
0
templates/adding_a_new_model/tests/test_tokenization_xxx.py
Executable file → Normal file
0
templates/adding_a_new_model/tokenization_xxx.py
Executable file → Normal file
0
templates/adding_a_new_model/tokenization_xxx.py
Executable file → Normal file
Reference in New Issue
Block a user