From a26528b897930679707a238bab773f84910f0669 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Danie=CC=88l=20de=20Kok?= Date: Thu, 21 Dec 2023 15:08:09 +0100 Subject: [PATCH] Fix merge fallout --- spacy/ml/models/textcat.py | 1 + 1 file changed, 1 insertion(+) diff --git a/spacy/ml/models/textcat.py b/spacy/ml/models/textcat.py index 61cd30f35..3e5471ab3 100644 --- a/spacy/ml/models/textcat.py +++ b/spacy/ml/models/textcat.py @@ -278,6 +278,7 @@ def _init_parametric_attention_with_residual_nonlinear(model, X, Y) -> Model: model.get_ref("norm_layer").set_dim("nI", tok2vec_width) model.get_ref("norm_layer").set_dim("nO", tok2vec_width) init_chain(model, X, Y) + return model @registry.architectures("spacy.TextCatReduce.v1")