smaller

Author: v-chen_data

tests/test_full_nlp.py

@@ -265,10 +265,10 @@ def test_full_nlp_pipeline(
         # Updated minimal BERT configuration
         config = BertConfig(
             vocab_size=30522,
-            hidden_size=64,
+            hidden_size=16,
             num_hidden_layers=2,
             num_attention_heads=2,
-            intermediate_size=256,
+            intermediate_size=64,
             num_labels=3,
         )
         tiny_bert_model = BertForMaskedLM(config)