Merge pull request flairNLP#1337 from flairNLP/flairNLPGH-1270-pooled…

…-embeddings flairNLPGH-1270: fix memory issue in PooledFlairEmbeddings
deaputri · Jan 7, 2020 · acc4473 · acc4473
2 parents 1626111 + 89a14c9
commit acc4473
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/flair/embeddings.py b/flair/embeddings.py
@@ -2144,7 +2144,6 @@ def _add_embeddings_internal(self, sentences: List[Sentence]) -> List[Sentence]:
 
                 # update embedding
                 local_embedding = token._embeddings[self.context_embeddings.name]
-                local_embedding = local_embedding.to(flair.device)
 
                 # check token.text is empty or not
                 if token.text: