From b59b924e497ac0674d897e08ee99541ff62d7335 Mon Sep 17 00:00:00 2001
From: Paul O'Leary McCann <polm@dampfkraft.com>
Date: Wed, 6 Jul 2022 19:22:19 +0900
Subject: [PATCH] Use normal PyTorchWrapper in coref

---
 spacy/ml/models/coref.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/spacy/ml/models/coref.py b/spacy/ml/models/coref.py
index 2b38beef7..7ad493ac0 100644
--- a/spacy/ml/models/coref.py
+++ b/spacy/ml/models/coref.py
@@ -62,7 +62,6 @@ def coref_init(model: Model, X=None, Y=None):
     antecedent_batch_size = model.attrs["antecedent_batch_size"]
     distance_embedding_size = model.attrs["distance_embedding_size"]
 
-    PyTorchWrapper = registry.get("layers", "PyTorchWrapper.v2")
     model._layers = [
         PyTorchWrapper(
             CorefClusterer(