Fix flaky test_batching_equivalence (#35564)

* yes! * oh no!!! * oh no!!! * style * oh no!!! * oh no!!! * oh no!!! * oh no!!! --------- Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
2025-01-09 14:00:08 +01:00
parent 4adc415b6d
commit 1b2f942af7
3 changed files with 34 additions and 6 deletions
--- a/tests/models/upernet/test_modeling_upernet.py
+++ b/tests/models/upernet/test_modeling_upernet.py
@@ -82,7 +82,6 @@ class UperNetModelTester:
        self.out_features = out_features
        self.num_labels = num_labels
        self.scope = scope
-        self.num_hidden_layers = num_stages

    def prepare_config_and_inputs(self):
        pixel_values = floats_tensor([self.batch_size, self.num_channels, self.image_size, self.image_size])