speed up test (#18106)

d4ebd4e1 · Sijun He · GitHub · b7d8bd37 · d4ebd4e1
Unverified Commit d4ebd4e1 authored Jul 12, 2022 by Sijun He Committed by GitHub Jul 12, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

tests/models/nezha/test_modeling_nezha.py tests/models/nezha/test_modeling_nezha.py +2 -1

No files found.
--- 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,