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 PIL import Image
from ..worker.command import Progress
from ..errors import CancelledException, RetryException from ..errors import CancelledException, RetryException
from ..output import save_image from ..output import save_image
from ..params import ImageParams, Size, StageParams from ..params import ImageParams, Size, StageParams
from ..server import ServerContext from ..server import ServerContext
from ..utils import is_debug, run_gc from ..utils import is_debug, run_gc
from ..worker import ProgressCallback, WorkerContext from ..worker import ProgressCallback, WorkerContext
from ..worker.command import Progress
from .base import BaseStage from .base import BaseStage
from .result import StageResult from .result import StageResult
from .tile import needs_tile, process_tile_order 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 ..diffusers.utils import replace_wildcards
from ..output import make_job_name, make_output_names 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 ..transformers.run import run_txt2txt_pipeline
from ..utils import ( from ..utils import (
base_join, base_join,
@ -34,7 +34,7 @@ from ..utils import (
load_config_str, load_config_str,
sanitize_name, sanitize_name,
) )
from ..worker.command import JobStatus, JobType from ..worker.command import JobStatus, JobType, Progress
from ..worker.pool import DevicePoolExecutor from ..worker.pool import DevicePoolExecutor
from .context import ServerContext from .context import ServerContext
from .load import ( from .load import (