From 1b58d337e30b4c9397473324810ca7afa35e462e Mon Sep 17 00:00:00 2001 From: Pedro Cuenca Date: Mon, 22 May 2023 12:35:53 +0200 Subject: [PATCH] Style --- tests/pipelines/controlnet/test_controlnet.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/tests/pipelines/controlnet/test_controlnet.py b/tests/pipelines/controlnet/test_controlnet.py index fa400fefdf..203fe1e51d 100644 --- a/tests/pipelines/controlnet/test_controlnet.py +++ b/tests/pipelines/controlnet/test_controlnet.py @@ -15,11 +15,11 @@ import gc import tempfile +import traceback import unittest import numpy as np import torch -import traceback from packaging import version from transformers import CLIPTextConfig, CLIPTextModel, CLIPTokenizer @@ -640,14 +640,12 @@ class ControlNetPipelineSlowTests(unittest.TestCase): expected_slice = np.array([0.2724, 0.2846, 0.2724, 0.3843, 0.3682, 0.2736, 0.4675, 0.3862, 0.2887]) assert np.abs(image_slice.flatten() - expected_slice).max() < 1e-2 - def test_stable_diffusion_compile(self): if version.parse(torch.__version__) < version.parse("2.0"): print(f"Test `test_stable_diffusion_compile` is skipped because {torch.__version__} is < 2.0") return run_test_in_subprocess(test_case=self, target_func=_test_stable_diffusion_compile, inputs=None) - def test_v11_shuffle_global_pool_conditions(self): controlnet = ControlNetModel.from_pretrained("lllyasviel/control_v11e_sd15_shuffle")