From 7b88371832f075ba431c497b89ba4c3e7ea1ffd4 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sat, 4 Mar 2023 22:13:09 -0600 Subject: [PATCH] lint(api): use load pattern in server submodule --- api/onnx_web/main.py | 2 +- api/onnx_web/server/api.py | 4 ++-- api/onnx_web/server/{config.py => load.py} | 0 api/onnx_web/server/params.py | 2 +- 4 files changed, 4 insertions(+), 4 deletions(-) rename api/onnx_web/server/{config.py => load.py} (100%) diff --git a/api/onnx_web/main.py b/api/onnx_web/main.py index 7b8ea325..f41667cd 100644 --- a/api/onnx_web/main.py +++ b/api/onnx_web/main.py @@ -10,7 +10,7 @@ from setproctitle import setproctitle from torch.multiprocessing import set_start_method from .server.api import register_api_routes -from .server.config import ( +from .server.load import ( get_available_platforms, load_models, load_params, diff --git a/api/onnx_web/server/api.py b/api/onnx_web/server/api.py index ed70c76f..76879641 100644 --- a/api/onnx_web/server/api.py +++ b/api/onnx_web/server/api.py @@ -19,7 +19,7 @@ from ..diffusion.run import ( from ..image import valid_image # mask filters; noise sources from ..output import json_params, make_output_name from ..params import Border, StageParams, TileOrder, UpscaleParams -from ..transformers import run_txt2txt_pipeline +from ..transformers.run import run_txt2txt_pipeline from ..utils import ( base_join, get_and_clamp_float, @@ -31,7 +31,7 @@ from ..utils import ( sanitize_name, ) from ..worker.pool import DevicePoolExecutor -from .config import ( +from .load import ( get_available_platforms, get_config_params, get_config_value, diff --git a/api/onnx_web/server/config.py b/api/onnx_web/server/load.py similarity index 100% rename from api/onnx_web/server/config.py rename to api/onnx_web/server/load.py diff --git a/api/onnx_web/server/params.py b/api/onnx_web/server/params.py index b9bc16b5..ca4c69c2 100644 --- a/api/onnx_web/server/params.py +++ b/api/onnx_web/server/params.py @@ -7,7 +7,7 @@ from flask import request from ..diffusion.load import pipeline_schedulers from ..params import Border, DeviceParams, ImageParams, Size, UpscaleParams from ..utils import get_and_clamp_float, get_and_clamp_int, get_from_list, get_not_empty -from .config import ( +from .load import ( get_available_platforms, get_config_value, get_correction_models,