Make more test models smaller (#25005)

* Make more test models tiny * Make more test models tiny * More models * More models
2023-07-24 10:08:47 -04:00
parent 8f1f0bf50f
commit 42571f6eb8
22 changed files with 149 additions and 137 deletions
--- a/tests/models/ctrl/test_modeling_tf_ctrl.py
+++ b/tests/models/ctrl/test_modeling_tf_ctrl.py
@@ -95,7 +95,7 @@ class TFCTRLModelTester(object):
            n_embd=self.hidden_size,
            n_layer=self.num_hidden_layers,
            n_head=self.num_attention_heads,
-            # intermediate_size=self.intermediate_size,
+            dff=self.intermediate_size,
            # hidden_act=self.hidden_act,
            # hidden_dropout_prob=self.hidden_dropout_prob,
            # attention_probs_dropout_prob=self.attention_probs_dropout_prob,