fix tests

a-r-r-o-w · a-r-r-o-w · commit 221b486cd041 · 2024-10-14T14:10:08.000+02:00
diff --git a/tests/models/transformers/test_models_transformer_cogview3plus.py b/tests/models/transformers/test_models_transformer_cogview3plus.py
@@ -78,7 +78,6 @@ def prepare_init_args_and_inputs_for_common(self):
             "text_embed_dim": 8,
             "time_embed_dim": 8,
             "condition_dim": 2,
-            "pooled_projection_dim": 12,
             "pos_embed_max_size": 8,
             "sample_size": 8,
         }
diff --git a/tests/pipelines/cogview3/test_cogview3plus.py b/tests/pipelines/cogview3/test_cogview3plus.py
@@ -69,7 +69,6 @@ def get_dummy_components(self):
             text_embed_dim=32,  # Must match with tiny-random-t5
             time_embed_dim=8,
             condition_dim=2,
-            pooled_projection_dim=12,
             pos_embed_max_size=8,
             sample_size=8,
         )

Original file line number	Diff line number	Diff line change
`@@ -78,7 +78,6 @@ def prepare_init_args_and_inputs_for_common(self):`
`78`	`78`	`"text_embed_dim": 8,`
`79`	`79`	`"time_embed_dim": 8,`
`80`	`80`	`"condition_dim": 2,`
`81`		`- "pooled_projection_dim": 12,`
`82`	`81`	`"pos_embed_max_size": 8,`
`83`	`82`	`"sample_size": 8,`
`84`	`83`	`}`
Original file line number	Diff line number	Diff line change
`@@ -69,7 +69,6 @@ def get_dummy_components(self):`
`69`	`69`	`text_embed_dim=32, # Must match with tiny-random-t5`
`70`	`70`	`time_embed_dim=8,`
`71`	`71`	`condition_dim=2,`
`72`		`- pooled_projection_dim=12,`
`73`	`72`	`pos_embed_max_size=8,`
`74`	`73`	`sample_size=8,`
`75`	`74`	`)`