1
0
Fork 0

lint(api): remove unused imports

This commit is contained in:
Sean Sube 2023-02-05 15:33:56 -06:00
parent 27da3c3334
commit cb3590720b
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
4 changed files with 1 additions and 6 deletions

View File

@ -1,6 +1,5 @@
from logging import getLogger
import numpy as np
import torch
from diffusers import OnnxStableDiffusionPipeline
from PIL import Image

View File

@ -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.*"

View File

@ -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

View File

@ -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