Skip to content

Commit 496e967

Browse files
authored
Disable DINO quantisation checks (#2397)
1 parent 6423e06 commit 496e967

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

keras_hub/src/models/dinov2/dinov2_backbone_test.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,7 @@ def test_backbone_basics(self):
3535
init_kwargs=self.init_kwargs,
3636
input_data=self.input_data,
3737
expected_output_shape=(2, sequence_length, hidden_dim),
38+
run_quantization_check=False,
3839
)
3940

4041
@pytest.mark.large
@@ -126,6 +127,7 @@ def test_backbone_basics(self):
126127
init_kwargs=self.init_kwargs,
127128
input_data=self.input_data,
128129
expected_output_shape=(2, sequence_length, hidden_dim),
130+
run_quantization_check=False,
129131
)
130132

131133
@pytest.mark.large

0 commit comments

Comments
 (0)