Skip to content

GitLab

Unverified Commit cd813499 authored by Dhruv Nair's avatar Dhruv Nair Committed by GitHub
Browse files

[CI] Skip ONNX Upscale tests (#11774) 

update
parent fbddf028
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
tests/pipelines/stable_diffusion/test_onnx_stable_diffusion_upscale.py
View file @ cd813499
...@@ -42,6 +42,10 @@ if is_onnx_available(): ...@@ -42,6 +42,10 @@ if is_onnx_available():
import onnxruntime as ort 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): class OnnxStableDiffusionUpscalePipelineFastTests(OnnxPipelineTesterMixin, unittest.TestCase):
# TODO: is there an appropriate internal test set? # TODO: is there an appropriate internal test set?
hub_checkpoint = "ssube/stable-diffusion-x4-upscaler-onnx" hub_checkpoint = "ssube/stable-diffusion-x4-upscaler-onnx"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment