Moved the encoded_prompts to correct device

2019-12-31 13:11:41 +01:00
parent 1efc208ff3
commit dd4df80f0b
285 changed files with 1 additions and 0 deletions
--- a/examples/README.md
+++ b/examples/README.md
--- a/examples/benchmarks.py
+++ b/examples/benchmarks.py
--- a/examples/contrib/README.md
+++ b/examples/contrib/README.md
--- a/examples/contrib/run_camembert.py
+++ b/examples/contrib/run_camembert.py
--- a/examples/contrib/run_openai_gpt.py
+++ b/examples/contrib/run_openai_gpt.py
--- a/examples/contrib/run_swag.py
+++ b/examples/contrib/run_swag.py
--- a/examples/contrib/run_transfo_xl.py
+++ b/examples/contrib/run_transfo_xl.py
--- a/examples/distillation/README.md
+++ b/examples/distillation/README.md
--- a/examples/distillation/distiller.py
+++ b/examples/distillation/distiller.py
--- a/examples/distillation/grouped_batch_sampler.py
+++ b/examples/distillation/grouped_batch_sampler.py
--- a/examples/distillation/lm_seqs_dataset.py
+++ b/examples/distillation/lm_seqs_dataset.py
--- a/examples/distillation/requirements.txt
+++ b/examples/distillation/requirements.txt
--- a/examples/distillation/run_squad_w_distillation.py
+++ b/examples/distillation/run_squad_w_distillation.py
--- a/examples/distillation/scripts/binarized_data.py
+++ b/examples/distillation/scripts/binarized_data.py
--- a/examples/distillation/scripts/extract.py
+++ b/examples/distillation/scripts/extract.py
--- a/examples/distillation/scripts/extract_distilbert.py
+++ b/examples/distillation/scripts/extract_distilbert.py
--- 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
--- a/examples/run_squad.py
+++ b/examples/run_squad.py
--- a/examples/run_tf_glue.py
+++ b/examples/run_tf_glue.py
--- a/examples/run_tf_ner.py
+++ b/examples/run_tf_ner.py
--- a/examples/run_xnli.py
+++ b/examples/run_xnli.py
--- a/examples/summarization/README.md
+++ b/examples/summarization/README.md
--- a/examples/summarization/configuration_bertabs.py
+++ b/examples/summarization/configuration_bertabs.py
--- a/examples/summarization/convert_bertabs_original_pytorch_checkpoint.py
+++ b/examples/summarization/convert_bertabs_original_pytorch_checkpoint.py
--- a/examples/summarization/modeling_bertabs.py
+++ b/examples/summarization/modeling_bertabs.py
--- a/examples/summarization/requirements.txt
+++ b/examples/summarization/requirements.txt
--- a/examples/summarization/run_summarization.py
+++ b/examples/summarization/run_summarization.py
--- a/examples/summarization/test_utils_summarization.py
+++ b/examples/summarization/test_utils_summarization.py
--- a/examples/summarization/utils_summarization.py
+++ b/examples/summarization/utils_summarization.py
--- a/examples/test_examples.py
+++ b/examples/test_examples.py
--- a/examples/tests_samples/.gitignore
+++ b/examples/tests_samples/.gitignore
--- a/examples/tests_samples/MRPC/dev.tsv
+++ b/examples/tests_samples/MRPC/dev.tsv
--- a/examples/tests_samples/MRPC/train.tsv
+++ b/examples/tests_samples/MRPC/train.tsv
--- a/examples/tests_samples/SQUAD/dev-v2.0.json
+++ b/examples/tests_samples/SQUAD/dev-v2.0.json
--- a/examples/tests_samples/SQUAD/train-v2.0.json
+++ b/examples/tests_samples/SQUAD/train-v2.0.json
--- a/examples/utils_multiple_choice.py
+++ b/examples/utils_multiple_choice.py
--- a/examples/utils_ner.py
+++ b/examples/utils_ner.py