From 505f286515c827ef04e8aec980202b32d33fd297 Mon Sep 17 00:00:00 2001
From: richardpaulhudson <richard@explosion.ai>
Date: Thu, 12 Jan 2023 16:18:16 +0100
Subject: [PATCH] Add dropout layer

---
 spacy/ml/models/tok2vec.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/spacy/ml/models/tok2vec.py b/spacy/ml/models/tok2vec.py
index f5cb72ee4..164990baf 100644
--- a/spacy/ml/models/tok2vec.py
+++ b/spacy/ml/models/tok2vec.py
@@ -1,4 +1,3 @@
-from encodings import search_function
 from typing import Optional, List, Union, cast
 import warnings
 from spacy.ml.richfeatureextractor import RichFeatureExtractor
@@ -6,6 +5,7 @@ from thinc.types import Floats2d, Ints2d, Ragged
 from thinc.api import chain, clone, concatenate, with_array, with_padded
 from thinc.api import Model, noop, list2ragged, ragged2list, HashEmbed
 from thinc.api import expand_window, residual, Maxout, Mish, PyTorchLSTM
+from thinc.api import Dropout
 
 from ...tokens import Doc
 from ...util import registry
@@ -296,6 +296,7 @@ def RichMultiHashEmbed(
             ),
             max_out,
             ragged2list(),
+            Dropout(0.0)
         )
     else:
         model = chain(
@@ -304,6 +305,7 @@ def RichMultiHashEmbed(
             with_array(concatenate(*embeddings)),
             max_out,
             ragged2list(),
+            Dropout(0.0)
         )
     return model