CI with num_hidden_layers=2 🚀🚀🚀 (#25266)
* CI with layers=2 --------- Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
@@ -49,7 +49,7 @@ class CpmAntModelTester:
|
||||
use_mc_token_ids=False,
|
||||
vocab_size=99,
|
||||
hidden_size=32,
|
||||
num_hidden_layers=3,
|
||||
num_hidden_layers=2,
|
||||
num_attention_heads=4,
|
||||
intermediate_size=37,
|
||||
num_buckets=32,
|
||||
|
||||
Reference in New Issue
Block a user