explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-07-18 12:12:20 +03:00
Code Issues Packages Projects Releases Wiki Activity

Hack for transformer listener size

Browse Source
This commit is contained in:
Paul O'Leary McCann 2022-03-16 15:15:53 +09:00
parent 7811a1194b
commit 6974f55daa
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
spacy/ml/models/coref.py
View File

@ -32,7 +32,12 @@ def build_wl_coref_model(
# span predictor embeddings # span predictor embeddings
sp_embedding_size: int = 64, sp_embedding_size: int = 64,
): ):
dim = tok2vec.get_dim("nO") # TODO fix this
try:
dim = tok2vec.get_dim("nO")
except ValueError:
# happens with transformer listener
dim = 768
with Model.define_operators({">>": chain}): with Model.define_operators({">>": chain}):
# TODO chain tok2vec with these models # TODO chain tok2vec with these models