diff --git a/ai_edge_torch/generative/test/test_quantize.py b/ai_edge_torch/generative/test/test_quantize.py index c2539bdd..2b81d3b8 100644 --- a/ai_edge_torch/generative/test/test_quantize.py +++ b/ai_edge_torch/generative/test/test_quantize.py @@ -160,6 +160,7 @@ def test_quantize_convert_toy_weight_sharing(self): ) def test_quantize_convert_toy_blockwise(self): + self.skipTest("b/448668892") config = toy_model.get_model_config() pytorch_model = toy_model.ToySingleLayerModel(config) idx = torch.unsqueeze(torch.arange(0, 100, dtype=torch.int), 0) @@ -176,6 +177,7 @@ def test_quantize_convert_toy_blockwise(self): ) def test_unsupported_block_size(self): + self.skipTest("b/448668892") config = toy_model.get_model_config() pytorch_model = toy_model.ToySingleLayerModel(config) idx = torch.unsqueeze(torch.arange(0, 100, dtype=torch.int), 0)