From 3037c2d45e709d34f43862c0efa2dcfe106eb0f1 Mon Sep 17 00:00:00 2001 From: DN6 Date: Mon, 23 Jun 2025 10:42:56 +0530 Subject: [PATCH] update --- .../stable_diffusion/test_onnx_stable_diffusion_upscale.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tests/pipelines/stable_diffusion/test_onnx_stable_diffusion_upscale.py b/tests/pipelines/stable_diffusion/test_onnx_stable_diffusion_upscale.py index 2df64ad1d6..e25118575f 100644 --- a/tests/pipelines/stable_diffusion/test_onnx_stable_diffusion_upscale.py +++ b/tests/pipelines/stable_diffusion/test_onnx_stable_diffusion_upscale.py @@ -42,6 +42,10 @@ if is_onnx_available(): import onnxruntime as ort +# TODO: (Dhruv) Update hub_checkpoint repo_id +@unittest.skip( + "There is a potential backdoor vulnerability in the hub_checkpoint. Skip running this test until resolved" +) class OnnxStableDiffusionUpscalePipelineFastTests(OnnxPipelineTesterMixin, unittest.TestCase): # TODO: is there an appropriate internal test set? hub_checkpoint = "ssube/stable-diffusion-x4-upscaler-onnx"