Skip to content

GitLab

Unverified Commit 9d3ff079 authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

fix tests (#3614)

parent a359ab4e
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 3 deletions
tests/pipelines/stable_diffusion/test_stable_diffusion.py
View file @ 9d3ff079
...@@ -41,7 +41,6 @@ from diffusers.models.attention_processor import AttnProcessor, LoRAXFormersAttn ...@@ -41,7 +41,6 @@ from diffusers.models.attention_processor import AttnProcessor, LoRAXFormersAttn
from diffusers.utils import load_numpy, nightly, slow, torch_device from diffusers.utils import load_numpy, nightly, slow, torch_device
from diffusers.utils.testing_utils import ( from diffusers.utils.testing_utils import (
CaptureLogger, CaptureLogger,
disable_full_determinism,
enable_full_determinism, enable_full_determinism,
require_torch_2, require_torch_2,
require_torch_gpu, require_torch_gpu,
...@@ -370,7 +369,7 @@ class StableDiffusionPipelineFastTests(PipelineLatentTesterMixin, PipelineTester ...@@ -370,7 +369,7 @@ class StableDiffusionPipelineFastTests(PipelineLatentTesterMixin, PipelineTester
@unittest.skipIf(not torch.cuda.is_available(), reason="xformers requires cuda") @unittest.skipIf(not torch.cuda.is_available(), reason="xformers requires cuda")
def test_stable_diffusion_attn_processors(self): def test_stable_diffusion_attn_processors(self):
disable_full_determinism() # disable_full_determinism()
device = "cuda" # ensure determinism for the device-dependent torch.Generator device = "cuda" # ensure determinism for the device-dependent torch.Generator
components = self.get_dummy_components() components = self.get_dummy_components()
sd_pipe = StableDiffusionPipeline(**components) sd_pipe = StableDiffusionPipeline(**components)
...@@ -416,7 +415,7 @@ class StableDiffusionPipelineFastTests(PipelineLatentTesterMixin, PipelineTester ...@@ -416,7 +415,7 @@ class StableDiffusionPipelineFastTests(PipelineLatentTesterMixin, PipelineTester
image = sd_pipe(**inputs).images image = sd_pipe(**inputs).images
assert image.shape == (1, 64, 64, 3) assert image.shape == (1, 64, 64, 3)
enable_full_determinism() # enable_full_determinism()
def test_stable_diffusion_no_safety_checker(self): def test_stable_diffusion_no_safety_checker(self):
pipe = StableDiffusionPipeline.from_pretrained( pipe = StableDiffusionPipeline.from_pretrained(
......
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