From a05fad8dcee87087368ad996fe2d76599b406e34 Mon Sep 17 00:00:00 2001 From: thomwolf Date: Thu, 11 Apr 2019 13:16:17 +0200 Subject: [PATCH] fix typo --- pytorch_pretrained_bert/modeling_openai.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_pretrained_bert/modeling_openai.py b/pytorch_pretrained_bert/modeling_openai.py index feae95d962..1a2a3feb20 100644 --- a/pytorch_pretrained_bert/modeling_openai.py +++ b/pytorch_pretrained_bert/modeling_openai.py @@ -608,7 +608,7 @@ class OpenAIGPTModel(OpenAIGPTPreTrainedModel): # Build new embeddings and initialize all new embeddings (in particular the special tokens) old_embed = self.tokens_embed self.tokens_embed = nn.Embedding(self.config.total_tokens_embeddings, self.config.n_embd) - self.tokens_embed.to(old_embed.device.weight.device) + self.tokens_embed.to(old_embed.weight.device) self.init_weights(self.tokens_embed) # Copy word embeddings from the previous weights self.tokens_embed.weight.data[:self.config.vocab_size, :] = old_embed.weight.data[:self.config.vocab_size, :]