[workflow] fixed oom tests (#5275)

* [workflow] fixed oom tests * polish * polish * polish
2025-09-06 11:32:10 +00:00 · 2024-01-16 18:55:13 +08:00
parent 04244aaaf1
commit d69cd2eb89
19 changed files with 50 additions and 582 deletions
--- a/tests/kit/model_zoo/transformers/gptj.py
+++ b/tests/kit/model_zoo/transformers/gptj.py
@@ -63,6 +63,9 @@ config = transformers.GPTJConfig(
    n_layer=2,
    n_head=4,
    vocab_size=50258,
+    n_embd=256,
+    hidden_size=256,
+    n_positions=512,
    attn_pdrop=0,
    embd_pdrop=0,
    resid_pdrop=0,