Skip to content

Commit 8e4b560

Browse files
authored
skip invalid test case for helios pipeline (#13218)
* skip invalid test case for helio pipeline Signed-off-by: Liu, Kaixuan <kaixuan.liu@intel.com> * update skip reason Signed-off-by: Liu, Kaixuan <kaixuan.liu@intel.com> --------- Signed-off-by: Liu, Kaixuan <kaixuan.liu@intel.com>
1 parent c6f72ad commit 8e4b560

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

tests/pipelines/helios/test_helios.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -139,9 +139,9 @@ def test_inference(self):
139139
generated_slice = torch.cat([generated_slice[:8], generated_slice[-8:]])
140140
self.assertTrue(torch.allclose(generated_slice, expected_slice, atol=1e-3))
141141

142-
# Override to set a more lenient max diff threshold.
142+
@unittest.skip("Helios uses a lot of mixed precision internally, which is not suitable for this test case")
143143
def test_save_load_float16(self):
144-
super().test_save_load_float16(expected_max_diff=0.03)
144+
pass
145145

146146
@unittest.skip("Test not supported")
147147
def test_attention_slicing_forward_pass(self):

0 commit comments

Comments
 (0)