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/distillation/README.md
Executable file → Normal file
0
examples/distillation/README.md
Executable file → Normal file
0
examples/distillation/distiller.py
Executable file → Normal file
0
examples/distillation/distiller.py
Executable file → Normal file
0
examples/distillation/grouped_batch_sampler.py
Executable file → Normal file
0
examples/distillation/grouped_batch_sampler.py
Executable file → Normal file
0
examples/distillation/lm_seqs_dataset.py
Executable file → Normal file
0
examples/distillation/lm_seqs_dataset.py
Executable file → Normal file
0
examples/distillation/requirements.txt
Executable file → Normal file
0
examples/distillation/requirements.txt
Executable file → Normal file
0
examples/distillation/run_squad_w_distillation.py
Executable file → Normal file
0
examples/distillation/run_squad_w_distillation.py
Executable file → Normal file
0
examples/distillation/scripts/binarized_data.py
Executable file → Normal file
0
examples/distillation/scripts/binarized_data.py
Executable file → Normal file
0
examples/distillation/scripts/extract.py
Executable file → Normal file
0
examples/distillation/scripts/extract.py
Executable file → Normal file
0
examples/distillation/scripts/extract_distilbert.py
Executable file → Normal file
0
examples/distillation/scripts/extract_distilbert.py
Executable file → Normal file
0
examples/distillation/scripts/token_counts.py
Executable file → Normal file
0
examples/distillation/scripts/token_counts.py
Executable file → Normal file
0
examples/distillation/train.py
Executable file → Normal file
0
examples/distillation/train.py
Executable file → Normal file
0
examples/distillation/training_configs/distilbert-base-uncased.json
Executable file → Normal file
0
examples/distillation/training_configs/distilbert-base-uncased.json
Executable file → Normal file
0
examples/distillation/training_configs/distilgpt2.json
Executable file → Normal file
0
examples/distillation/training_configs/distilgpt2.json
Executable file → Normal file
0
examples/distillation/utils.py
Executable file → Normal file
0
examples/distillation/utils.py
Executable file → Normal file
Reference in New Issue
Block a user