apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
api/onnx_web/chain/source_txt2img.py
View File

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

2
api/onnx_web/convert.py
View File

@ -20,8 +20,6 @@ from diffusers import (
from onnx import load, save_model from onnx import load, save_model
from torch.onnx import export from torch.onnx import export
from . import logging
# suppress common but harmless warnings, https://github.com/ssube/onnx-web/issues/75 # suppress common but harmless warnings, https://github.com/ssube/onnx-web/issues/75
warnings.filterwarnings( warnings.filterwarnings(
"ignore", ".*The shape inference of prim::Constant type is missing.*" "ignore", ".*The shape inference of prim::Constant type is missing.*"

1
api/onnx_web/diffusion/run.py
View File

@ -1,7 +1,6 @@
from logging import getLogger from logging import getLogger
from typing import Any from typing import Any
import numpy as np
import torch import torch
from diffusers import OnnxStableDiffusionImg2ImgPipeline, OnnxStableDiffusionPipeline from diffusers import OnnxStableDiffusionImg2ImgPipeline, OnnxStableDiffusionPipeline
from PIL import Image, ImageChops from PIL import Image, ImageChops

3
api/onnx_web/serve.py
View File

@ -29,7 +29,6 @@ from jsonschema import validate
from onnxruntime import get_available_providers from onnxruntime import get_available_providers
from PIL import Image from PIL import Image
from . import logging
from .chain import ( from .chain import (
ChainPipeline, ChainPipeline,
blend_img2img, blend_img2img,
@ -168,7 +167,7 @@ def pipeline_from_request() -> Tuple[DeviceParams, ImageParams, Size]:
device = available_platforms[0] device = available_platforms[0]
if device is None: if device is None:
logger.warn('unknown platform: %s', device_name) logger.warn("unknown platform: %s", device_name)
device = available_platforms[0] device = available_platforms[0]
# pipeline stuff # pipeline stuff