lint(api): remove unused imports
This commit is contained in:
parent
27da3c3334
commit
cb3590720b
|
@ -1,6 +1,5 @@
|
|||
from logging import getLogger
|
||||
|
||||
import numpy as np
|
||||
import torch
|
||||
from diffusers import OnnxStableDiffusionPipeline
|
||||
from PIL import Image
|
||||
|
|
|
@ -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.*"
|
||||
|
|
|
@ -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
|
||||
|
|
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue