Fix load_vectors_into_model function

This commit is contained in:
Matthew Honnibal 2020-07-28 21:59:51 +02:00
parent 7299419fe4
commit 7852a68a75
1 changed files with 2 additions and 2 deletions

View File

@ -202,8 +202,8 @@ def load_vectors_into_model(
# I guess we should add the strings from the vectors_nlp model?
# E.g. if someone does a similarity query, they might expect the strings.
for key in nlp.vocab.vectors.key2row:
if key in vectors_nlp.strings:
nlp.vocab.strings.add(vectors_nlp.strings[key])
if key in vectors_nlp.vocab.strings:
nlp.vocab.strings.add(vectors_nlp.vocab.strings[key])
def load_model(