From 5d7e84571250b95ba689aae46315e495ebdb918b Mon Sep 17 00:00:00 2001
From: thomwolf <thomwolf@gmail.com>
Date: Fri, 8 Feb 2019 11:08:43 +0100
Subject: [PATCH] fix model on cuda

---
 examples/run_openai_gpt.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/examples/run_openai_gpt.py b/examples/run_openai_gpt.py
index 9b6383fc08..b3410cb425 100644
--- a/examples/run_openai_gpt.py
+++ b/examples/run_openai_gpt.py
@@ -135,6 +135,7 @@ def main():
     tokenizer = OpenAIGPTTokenizer.from_pretrained(args.model_name, special_tokens=special_tokens)
     special_tokens_ids = list(tokenizer.convert_tokens_to_ids(token) for token in special_tokens)
     model = OpenAIGPTDoubleHeadsModel.from_pretrained(args.model_name, num_special_tokens=len(special_tokens))
+    model.to(device)
 
     # Load and encode the datasets
     def tokenize_and_encode(obj):