apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix imports

This commit is contained in:
Sean Sube 2024-01-04 19:16:40 -06:00
parent b6da935be6
commit 9a1389d03b
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
api/onnx_web/chain/pipeline.py
View File

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

4
api/onnx_web/server/api.py
View File

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