speed up test (#18106)

2022-07-12 16:28:28 +08:00
parent b7d8bd378c
commit d4ebd4e112
1 changed files with 2 additions and 1 deletions
--- a/tests/models/nezha/test_modeling_nezha.py
+++ b/tests/models/nezha/test_modeling_nezha.py
@@ -60,7 +60,8 @@ class NezhaModelTester:
        hidden_act="gelu",
        hidden_dropout_prob=0.1,
        attention_probs_dropout_prob=0.1,
-        max_position_embeddings=512,
+        max_position_embeddings=128,
+        max_relative_position=32,
        type_vocab_size=16,
        type_sequence_label_size=2,
        initializer_range=0.02,