flairNLPGH-1270: fix memory issue in PooledFlairEmbeddings

deaputri · Jan 7, 2020 · 89a14c9 · 89a14c9
1 parent 1626111
commit 89a14c9
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: