diff --git a/flair/embeddings/token.py b/flair/embeddings/token.py index b4e3edc1fd..76173bac80 100644 --- a/flair/embeddings/token.py +++ b/flair/embeddings/token.py @@ -204,7 +204,7 @@ def __init__( self.__embedding_length: int = precomputed_word_embeddings.vector_size - vectors = np.row_stack( + vectors = np.vstack( ( precomputed_word_embeddings.vectors, np.zeros(self.__embedding_length, dtype="float"), @@ -399,7 +399,7 @@ def __setstate__(self, state: Dict[str, Any]): state.setdefault("field", None) if "precomputed_word_embeddings" in state: precomputed_word_embeddings: KeyedVectors = state.pop("precomputed_word_embeddings") - vectors = np.row_stack( + vectors = np.vstack( ( precomputed_word_embeddings.vectors, np.zeros(precomputed_word_embeddings.vector_size, dtype="float"),