Moved the encoded_prompts to correct device

dd4df80f · alberduris · Lysandre Debut · 1efc208f · dd4df80f · dd4df80f
Commit dd4df80f authored Dec 31, 2019 by alberduris Committed by Lysandre Debut Jan 06, 2020
20 changed files
--- a/examples/distillation/scripts/token_counts.py
+++ b/examples/distillation/scripts/token_counts.py
--- a/examples/distillation/train.py
+++ b/examples/distillation/train.py
--- a/examples/distillation/training_configs/distilbert-base-uncased.json
+++ b/examples/distillation/training_configs/distilbert-base-uncased.json
--- a/examples/distillation/training_configs/distilgpt2.json
+++ b/examples/distillation/training_configs/distilgpt2.json
--- a/examples/distillation/utils.py
+++ b/examples/distillation/utils.py
--- a/examples/mm-imdb/run_mmimdb.py
+++ b/examples/mm-imdb/run_mmimdb.py
--- a/examples/mm-imdb/utils_mmimdb.py
+++ b/examples/mm-imdb/utils_mmimdb.py
--- a/examples/pplm/README.md
+++ b/examples/pplm/README.md
--- a/examples/pplm/imgs/headfigure.png
+++ b/examples/pplm/imgs/headfigure.png
--- a/examples/pplm/imgs/wooly.png
+++ b/examples/pplm/imgs/wooly.png
--- a/examples/pplm/pplm_classification_head.py
+++ b/examples/pplm/pplm_classification_head.py
--- a/examples/pplm/run_pplm.py
+++ b/examples/pplm/run_pplm.py
--- a/examples/pplm/run_pplm_discrim_train.py
+++ b/examples/pplm/run_pplm_discrim_train.py
--- a/examples/requirements.txt
+++ b/examples/requirements.txt
--- a/examples/run_bertology.py
+++ b/examples/run_bertology.py
--- a/examples/run_generation.py
+++ b/examples/run_generation.py
@@ -212,6 +212,7 @@ def main():
        prepare_input = PREPROCESSING_FUNCTIONS.get(args.model_type)
        prompt_text = prepare_input(args, model, tokenizer, prompt_text)
    encoded_prompt = tokenizer.encode(prompt_text, add_special_tokens=False, return_tensors="pt")
+    encoded_prompt = encoded_prompt.to(args.device)
    output_sequences = model.generate(
        input_ids=encoded_prompt,

--- a/examples/run_glue.py
+++ b/examples/run_glue.py
--- a/examples/run_lm_finetuning.py
+++ b/examples/run_lm_finetuning.py
--- a/examples/run_multiple_choice.py
+++ b/examples/run_multiple_choice.py
--- a/examples/run_ner.py
+++ b/examples/run_ner.py