fix style

huggingface · Jul 11, 2023 · 00d26ba · 00d26ba
1 parent bac78eb
commit 00d26ba
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 6 deletions.
diff --git a/optimum/pipelines/diffusers/pipeline_stable_diffusion_img2img.py b/optimum/pipelines/diffusers/pipeline_stable_diffusion_img2img.py
@@ -14,17 +14,16 @@
 
 import inspect
 import logging
-import warnings
 from typing import Callable, List, Optional, Union
 
 import numpy as np
 import PIL
 import torch
 from diffusers.pipelines.stable_diffusion import StableDiffusionPipelineOutput
 from diffusers.utils import deprecate
-from .pipeline_utils import preprocess
 
 from .pipeline_stable_diffusion import StableDiffusionPipelineMixin
+from .pipeline_utils import preprocess
 
 
 logger = logging.getLogger(__name__)

diff --git a/optimum/pipelines/diffusers/pipeline_stable_diffusion_xl_img2img.py b/optimum/pipelines/diffusers/pipeline_stable_diffusion_xl_img2img.py
@@ -21,8 +21,7 @@
 import torch
 from diffusers.pipelines.stable_diffusion_xl import StableDiffusionXLPipelineOutput
 
-from .pipeline_utils import DiffusionPipelineMixin, rescale_noise_cfg, preprocess
-import warnings
+from .pipeline_utils import DiffusionPipelineMixin, preprocess, rescale_noise_cfg
 
 
 logger = logging.getLogger(__name__)

diff --git a/optimum/pipelines/diffusers/pipeline_utils.py b/optimum/pipelines/diffusers/pipeline_utils.py
@@ -19,9 +19,9 @@
 import PIL
 import torch
 from diffusers import ConfigMixin
+from diffusers.utils import PIL_INTERPOLATION
 from PIL import Image
 from tqdm.auto import tqdm
-from diffusers.utils import PIL_INTERPOLATION, deprecate
 
 
 class DiffusionPipelineMixin(ConfigMixin):

diff --git a/tests/onnxruntime/test_stable_diffusion_pipeline.py b/tests/onnxruntime/test_stable_diffusion_pipeline.py
@@ -336,7 +336,7 @@ def generate_inputs(self, height=128, width=128):
         return inputs
 
 
-class ORTStableDiffusionImg2ImgPipelineTest(ORTModelTestMixin):
+class ORTStableDiffusionXLImg2ImgPipelineTest(ORTModelTestMixin):
     SUPPORTED_ARCHITECTURES = [
         "stable-diffusion-xl",
     ]