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/pplm/README.md
Executable file → Normal file
0
examples/pplm/README.md
Executable file → Normal file
0
examples/pplm/imgs/headfigure.png
Executable file → Normal file
0
examples/pplm/imgs/headfigure.png
Executable file → Normal file
|
Before Width: | Height: | Size: 653 KiB After Width: | Height: | Size: 653 KiB |
0
examples/pplm/imgs/wooly.png
Executable file → Normal file
0
examples/pplm/imgs/wooly.png
Executable file → Normal file
|
Before Width: | Height: | Size: 664 KiB After Width: | Height: | Size: 664 KiB |
0
examples/pplm/pplm_classification_head.py
Executable file → Normal file
0
examples/pplm/pplm_classification_head.py
Executable file → Normal file
0
examples/pplm/run_pplm.py
Executable file → Normal file
0
examples/pplm/run_pplm.py
Executable file → Normal file
0
examples/pplm/run_pplm_discrim_train.py
Executable file → Normal file
0
examples/pplm/run_pplm_discrim_train.py
Executable file → Normal file
Reference in New Issue
Block a user