Moved the encoded_prompts to correct device
This commit is contained in:
committed by
Lysandre Debut
parent
1efc208ff3
commit
dd4df80f0b
0
examples/distillation/training_configs/distilbert-base-uncased.json
Normal file → Executable file
0
examples/distillation/training_configs/distilbert-base-uncased.json
Normal file → Executable file
0
examples/distillation/training_configs/distilgpt2.json
Normal file → Executable file
0
examples/distillation/training_configs/distilgpt2.json
Normal file → Executable file
Reference in New Issue
Block a user