From dab7894cf38cfcd883fd98c69476be85a0d8fd6b Mon Sep 17 00:00:00 2001 From: Paul O'Leary McCann Date: Tue, 27 Dec 2022 17:34:34 +0900 Subject: [PATCH] Add HashEmbedCNN to list of tok2vec architectures --- spacy/cli/configure.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/spacy/cli/configure.py b/spacy/cli/configure.py index 2d0f1ade2..77d0dfe68 100644 --- a/spacy/cli/configure.py +++ b/spacy/cli/configure.py @@ -10,7 +10,11 @@ from spacy.language import Language from ._util import configure_cli, Arg, Opt # These are the architectures that are recognized as tok2vec/feature sources. -TOK2VEC_ARCHS = [("spacy", "Tok2Vec"), ("spacy-transformers", "TransformerModel")] +TOK2VEC_ARCHS = [ + ("spacy", "Tok2Vec"), + ("spacy", "HashEmbedCNN"), + ("spacy-transformers", "TransformerModel"), +] # These are the listeners. LISTENER_ARCHS = [ ("spacy", "Tok2VecListener"),