Skip to content

Commit 9d3ff07

Browse files
fix tests (huggingface#3614)
1 parent a359ab4 commit 9d3ff07

File tree

2 files changed

+2
-3
lines changed

2 files changed

+2
-3
lines changed
File renamed without changes.

tests/pipelines/stable_diffusion/test_stable_diffusion.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,6 @@
4141
from diffusers.utils import load_numpy, nightly, slow, torch_device
4242
from diffusers.utils.testing_utils import (
4343
CaptureLogger,
44-
disable_full_determinism,
4544
enable_full_determinism,
4645
require_torch_2,
4746
require_torch_gpu,
@@ -370,7 +369,7 @@ def test_stable_diffusion_pndm(self):
370369

371370
@unittest.skipIf(not torch.cuda.is_available(), reason="xformers requires cuda")
372371
def test_stable_diffusion_attn_processors(self):
373-
disable_full_determinism()
372+
# disable_full_determinism()
374373
device = "cuda" # ensure determinism for the device-dependent torch.Generator
375374
components = self.get_dummy_components()
376375
sd_pipe = StableDiffusionPipeline(**components)
@@ -416,7 +415,7 @@ def test_stable_diffusion_attn_processors(self):
416415
image = sd_pipe(**inputs).images
417416
assert image.shape == (1, 64, 64, 3)
418417

419-
enable_full_determinism()
418+
# enable_full_determinism()
420419

421420
def test_stable_diffusion_no_safety_checker(self):
422421
pipe = StableDiffusionPipeline.from_pretrained(

0 commit comments

Comments
 (0)