From 9a1389d03b715bd6796b831865f33b2a6cec0a49 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Thu, 4 Jan 2024 19:16:40 -0600 Subject: [PATCH] fix imports --- api/onnx_web/chain/pipeline.py | 3 +-- api/onnx_web/server/api.py | 4 ++-- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/api/onnx_web/chain/pipeline.py b/api/onnx_web/chain/pipeline.py index a5f19830..67b046ab 100644 --- a/api/onnx_web/chain/pipeline.py +++ b/api/onnx_web/chain/pipeline.py @@ -5,14 +5,13 @@ from typing import Any, List, Optional, Tuple from PIL import Image -from ..worker.command import Progress - from ..errors import CancelledException, RetryException from ..output import save_image from ..params import ImageParams, Size, StageParams from ..server import ServerContext from ..utils import is_debug, run_gc from ..worker import ProgressCallback, WorkerContext +from ..worker.command import Progress from .base import BaseStage from .result import StageResult from .tile import needs_tile, process_tile_order diff --git a/api/onnx_web/server/api.py b/api/onnx_web/server/api.py index 266b35dc..02111afa 100644 --- a/api/onnx_web/server/api.py +++ b/api/onnx_web/server/api.py @@ -19,7 +19,7 @@ from ..diffusers.run import ( ) from ..diffusers.utils import replace_wildcards from ..output import make_job_name, make_output_names -from ..params import Progress, Size, StageParams, TileOrder +from ..params import Size, StageParams, TileOrder from ..transformers.run import run_txt2txt_pipeline from ..utils import ( base_join, @@ -34,7 +34,7 @@ from ..utils import ( load_config_str, sanitize_name, ) -from ..worker.command import JobStatus, JobType +from ..worker.command import JobStatus, JobType, Progress from ..worker.pool import DevicePoolExecutor from .context import ServerContext from .load import (