diff --git a/api/onnx_web/chain/source_txt2img.py b/api/onnx_web/chain/source_txt2img.py index 76072cce..9b2f8a0a 100644 --- a/api/onnx_web/chain/source_txt2img.py +++ b/api/onnx_web/chain/source_txt2img.py @@ -1,6 +1,5 @@ from logging import getLogger -import numpy as np import torch from diffusers import OnnxStableDiffusionPipeline from PIL import Image diff --git a/api/onnx_web/convert.py b/api/onnx_web/convert.py index eb9c6d97..a01991aa 100644 --- a/api/onnx_web/convert.py +++ b/api/onnx_web/convert.py @@ -20,8 +20,6 @@ from diffusers import ( from onnx import load, save_model from torch.onnx import export -from . import logging - # suppress common but harmless warnings, https://github.com/ssube/onnx-web/issues/75 warnings.filterwarnings( "ignore", ".*The shape inference of prim::Constant type is missing.*" diff --git a/api/onnx_web/diffusion/run.py b/api/onnx_web/diffusion/run.py index a9794091..961bb3c1 100644 --- a/api/onnx_web/diffusion/run.py +++ b/api/onnx_web/diffusion/run.py @@ -1,7 +1,6 @@ from logging import getLogger from typing import Any -import numpy as np import torch from diffusers import OnnxStableDiffusionImg2ImgPipeline, OnnxStableDiffusionPipeline from PIL import Image, ImageChops diff --git a/api/onnx_web/serve.py b/api/onnx_web/serve.py index 1ccdd8e8..0fb3f200 100644 --- a/api/onnx_web/serve.py +++ b/api/onnx_web/serve.py @@ -29,7 +29,6 @@ from jsonschema import validate from onnxruntime import get_available_providers from PIL import Image -from . import logging from .chain import ( ChainPipeline, blend_img2img, @@ -168,7 +167,7 @@ def pipeline_from_request() -> Tuple[DeviceParams, ImageParams, Size]: device = available_platforms[0] if device is None: - logger.warn('unknown platform: %s', device_name) + logger.warn("unknown platform: %s", device_name) device = available_platforms[0] # pipeline stuff