From c2ba71a2f5b9e885ca607664b78fe201faa1f72b Mon Sep 17 00:00:00 2001
From: calpt <calpt@mail.de>
Date: Tue, 8 Feb 2022 15:19:26 +0100
Subject: [PATCH] Fix T5 adapter test

---
 tests/test_adapter.py | 11 +++++------
 1 file changed, 5 insertions(+), 6 deletions(-)

diff --git a/tests/test_adapter.py b/tests/test_adapter.py
index b46a57e38..2a032b51f 100644
--- a/tests/test_adapter.py
+++ b/tests/test_adapter.py
@@ -369,12 +369,11 @@ class T5AdapterTestBase(AdapterTestBase):
     config = make_config(
         T5Config,
         d_model=16,
-        encoder_layers=2,
-        decoder_layers=2,
-        encoder_attention_heads=4,
-        decoder_attention_heads=4,
-        encoder_ffn_dim=4,
-        decoder_ffn_dim=4,
+        num_layers=2,
+        num_decoder_layers=2,
+        num_heads=4,
+        d_ff=4,
+        d_kv=16 // 4,
         tie_word_embeddings=False,
         decoder_start_token_id=0,
     )