From 2a6997c6eef8e341fb260914071848c7433498bf Mon Sep 17 00:00:00 2001 From: Huy Do Date: Wed, 19 Nov 2025 18:00:30 -0800 Subject: [PATCH 1/2] Skip stable-diffusion-2 on TorchBench CI Signed-off-by: Huy Do --- .ci/torchbench/install.sh | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/.ci/torchbench/install.sh b/.ci/torchbench/install.sh index e4e944e3e..95cb8e775 100644 --- a/.ci/torchbench/install.sh +++ b/.ci/torchbench/install.sh @@ -3,4 +3,6 @@ set -eux python3 -c "import torch; print(torch.__version__); print(torch.version.git_version)" -python3 install.py +# TODO (huydhn): Look for a more permanent solution once https://github.com/pytorch/pytorch/issues/167895 +# is resolved +python3 install.py --skip stable_diffusion_text_encoder stable_diffusion_unet From fe06547b9abb3870db8c5cb611cf12ca96ce94ee Mon Sep 17 00:00:00 2001 From: Huy Do Date: Fri, 21 Nov 2025 23:59:16 -0800 Subject: [PATCH 2/2] Skip tests too Signed-off-by: Huy Do --- test.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/test.py b/test.py index bb4884b6b..7ddc1ca48 100644 --- a/test.py +++ b/test.py @@ -181,6 +181,9 @@ def _load_tests(): # api, enable after PyTorch 1.13 release if "quantized" in path: continue + # https://github.com/pytorch/pytorch/issues/167895 + if "stable_diffusion" in path: + continue for device in devices: _load_test(path, device)