Fix T5 adapter test

adapter-hub · Feb 8, 2022 · c2ba71a · c2ba71a
1 parent 1e540d1
commit c2ba71a
Showing 1 changed file with 5 additions and 6 deletions.
diff --git 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,
  )