[tests] decorate StableDiffusion21PipelineSingleFileSlowTests with slow. (#7941)

decorate StableDiffusion21PipelineSingleFileSlowTests with slow.

[tests] decorate StableDiffusion21PipelineSingleFileSlowTests with slow. (#7941)
decorate StableDiffusion21PipelineSingleFileSlowTests with slow.
72780ff5 · Sayak Paul · GitHub · 69fdb872 · 72780ff5
Unverified Commit 72780ff5 authored May 15, 2024 by Sayak Paul Committed by GitHub May 14, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

tests/single_file/test_stable_diffusion_single_file.py tests/single_file/test_stable_diffusion_single_file.py +1 -0

No files found.
--- a/tests/single_file/test_stable_diffusion_single_file.py
+++ b/tests/single_file/test_stable_diffusion_single_file.py
@@ -82,6 +82,7 @@ class StableDiffusionPipelineSingleFileSlowTests(unittest.TestCase, SDSingleFile
        assert pipe.vae.config.scaling_factor == new_scaling_factor


+@slow
 class StableDiffusion21PipelineSingleFileSlowTests(unittest.TestCase, SDSingleFileTesterMixin):
    pipeline_class = StableDiffusionPipeline
    ckpt_path = "https://huggingface.co/stabilityai/stable-diffusion-2-1/blob/main/v2-1_768-ema-pruned.safetensors"