Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/vision.git" did not exist on "76d012454ed42be26d2e1b7897a4805c5b747420"
Unverified Commit 72780ff5 authored by Sayak Paul's avatar Sayak Paul Committed by GitHub
Browse files

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

decorate StableDiffusion21PipelineSingleFileSlowTests with slow.
parent 69fdb872
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
tests/single_file/test_stable_diffusion_single_file.py
View file @ 72780ff5
...@@ -82,6 +82,7 @@ class StableDiffusionPipelineSingleFileSlowTests(unittest.TestCase, SDSingleFile ...@@ -82,6 +82,7 @@ class StableDiffusionPipelineSingleFileSlowTests(unittest.TestCase, SDSingleFile
assert pipe.vae.config.scaling_factor == new_scaling_factor assert pipe.vae.config.scaling_factor == new_scaling_factor
@slow
class StableDiffusion21PipelineSingleFileSlowTests(unittest.TestCase, SDSingleFileTesterMixin): class StableDiffusion21PipelineSingleFileSlowTests(unittest.TestCase, SDSingleFileTesterMixin):
pipeline_class = StableDiffusionPipeline pipeline_class = StableDiffusionPipeline
ckpt_path = "https://huggingface.co/stabilityai/stable-diffusion-2-1/blob/main/v2-1_768-ema-pruned.safetensors" ckpt_path = "https://huggingface.co/stabilityai/stable-diffusion-2-1/blob/main/v2-1_768-ema-pruned.safetensors"
......
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