Avoid assuming encode.get_dim('nO') is set in tok2vec (explosion#6800)

honnibal · web-flow · commit ffc371350aa5 · 2021-01-24T14:37:33.000+11:00
diff --git a/spacy/ml/models/tok2vec.py b/spacy/ml/models/tok2vec.py
@@ -101,7 +101,8 @@ def build_Tok2Vec_model(
         embeddings, using an architecture such as a CNN, BiLSTM or transformer.
     """
     tok2vec = chain(embed, encode)
-    tok2vec.set_dim("nO", encode.get_dim("nO"))
+    if encode.has_dim("nO"):
+        tok2vec.set_dim("nO", encode.get_dim("nO"))
     tok2vec.set_ref("embed", embed)
     tok2vec.set_ref("encode", encode)
     return tok2vec
