1
0
Fork 0

apply lint

This commit is contained in:
Sean Sube 2023-04-01 12:06:31 -05:00
parent e0e09996de
commit 0f79f420b7
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
4 changed files with 26 additions and 4 deletions

View File

@ -11,7 +11,15 @@ from onnx_web.chain.utils import process_tile_order
from ..chain import blend_mask, upscale_outpaint
from ..chain.base import ChainProgress
from ..output import save_image, save_params
from ..params import Border, HighresParams, ImageParams, Size, StageParams, TileOrder, UpscaleParams
from ..params import (
Border,
HighresParams,
ImageParams,
Size,
StageParams,
TileOrder,
UpscaleParams,
)
from ..server import ServerContext
from ..utils import run_gc
from ..worker import WorkerContext

View File

@ -114,7 +114,9 @@ def save_params(
highres: Optional[HighresParams] = None,
) -> str:
path = base_join(ctx.output_path, f"{output}.json")
json = json_params(output, params, size, upscale=upscale, border=border, highres=highres)
json = json_params(
output, params, size, upscale=upscale, border=border, highres=highres
)
with open(path, "w") as f:
f.write(dumps(json))
logger.debug("saved image params to: %s", path)

View File

@ -44,7 +44,12 @@ from .load import (
get_noise_sources,
get_upscaling_models,
)
from .params import border_from_request, highres_from_request, pipeline_from_request, upscale_from_request
from .params import (
border_from_request,
highres_from_request,
pipeline_from_request,
upscale_from_request,
)
from .utils import wrap_route
logger = getLogger(__name__)

View File

@ -5,7 +5,14 @@ import numpy as np
from flask import request
from ..diffusers.load import pipeline_schedulers
from ..params import Border, DeviceParams, ImageParams, HighresParams, Size, UpscaleParams
from ..params import (
Border,
DeviceParams,
HighresParams,
ImageParams,
Size,
UpscaleParams,
)
from ..utils import get_and_clamp_float, get_and_clamp_int, get_from_list, get_not_empty
from .context import ServerContext
from .load import (