[test] smaller gpt2 test case

Hz188 · May 27, 2024 · df705a5 · df705a5
1 parent b96c639
commit df705a5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/kit/model_zoo/transformers/gpt.py b/tests/kit/model_zoo/transformers/gpt.py
@@ -121,7 +121,7 @@ def date_gen_for_double_heads():
 config = transformers.GPT2Config(
     n_layer=2,
     n_head=4,
-    n_embd=128,
+    n_embd=16,
     vocab_size=50258,
     attn_pdrop=0,
     embd_pdrop=0,