type lint
This commit is contained in:
parent
fab3ac68c2
commit
2b65077d82
|
@ -25,7 +25,7 @@ class TileCallback(Protocol):
|
||||||
Definition for a tile job function.
|
Definition for a tile job function.
|
||||||
"""
|
"""
|
||||||
|
|
||||||
def __call__(self, image: Image.Image, dims: Tuple[int, int, int]) -> StageResult:
|
def __call__(self, sources: List[Image.Image], mask: Image.Image, dims: Tuple[int, int, int]) -> StageResult:
|
||||||
"""
|
"""
|
||||||
Run this stage against a single tile.
|
Run this stage against a single tile.
|
||||||
"""
|
"""
|
||||||
|
@ -366,10 +366,7 @@ def process_tile_order(
|
||||||
scale: int,
|
scale: int,
|
||||||
filters: List[TileCallback],
|
filters: List[TileCallback],
|
||||||
**kwargs,
|
**kwargs,
|
||||||
) -> Image.Image:
|
) -> List[Image.Image]:
|
||||||
"""
|
|
||||||
TODO: needs to handle more than one image
|
|
||||||
"""
|
|
||||||
if order == TileOrder.grid:
|
if order == TileOrder.grid:
|
||||||
logger.debug("using grid tile order with tile size: %s", tile)
|
logger.debug("using grid tile order with tile size: %s", tile)
|
||||||
return process_tile_stack(
|
return process_tile_stack(
|
||||||
|
@ -483,7 +480,7 @@ def generate_tile_grid(
|
||||||
height: int,
|
height: int,
|
||||||
tile: int,
|
tile: int,
|
||||||
overlap: float = 0.0,
|
overlap: float = 0.0,
|
||||||
) -> List[Tuple[int, int]]:
|
) -> List[Tuple[int, int, Image.Image]]:
|
||||||
adj_tile = int(float(tile) * (1.0 - overlap))
|
adj_tile = int(float(tile) * (1.0 - overlap))
|
||||||
tiles_x = ceil(width / adj_tile)
|
tiles_x = ceil(width / adj_tile)
|
||||||
tiles_y = ceil(height / adj_tile)
|
tiles_y = ceil(height / adj_tile)
|
||||||
|
|
|
@ -1,5 +1,5 @@
|
||||||
from logging import getLogger
|
from logging import getLogger
|
||||||
from typing import Dict, Optional, Tuple
|
from typing import Dict, Optional, Tuple, Union
|
||||||
|
|
||||||
from flask import request
|
from flask import request
|
||||||
|
|
||||||
|
@ -345,7 +345,7 @@ PipelineParams = Tuple[Optional[DeviceParams], ImageParams, Size]
|
||||||
|
|
||||||
def pipeline_from_json(
|
def pipeline_from_json(
|
||||||
server: ServerContext,
|
server: ServerContext,
|
||||||
data: Dict[str, str],
|
data: Dict[str, Union[str, Dict[str, str]]],
|
||||||
default_pipeline: str = "txt2img",
|
default_pipeline: str = "txt2img",
|
||||||
) -> PipelineParams:
|
) -> PipelineParams:
|
||||||
"""
|
"""
|
||||||
|
|
|
@ -23,9 +23,14 @@ module = [
|
||||||
"basicsr",
|
"basicsr",
|
||||||
"boto3",
|
"boto3",
|
||||||
"codeformer",
|
"codeformer",
|
||||||
"codeformer.facelib.utils.misc",
|
"codeformer.basicsr",
|
||||||
"codeformer.facelib.utils",
|
"codeformer.basicsr.utils",
|
||||||
|
"codeformer.basicsr.utils.download_util",
|
||||||
|
"codeformer.basicsr.utils.registry",
|
||||||
"codeformer.facelib",
|
"codeformer.facelib",
|
||||||
|
"codeformer.facelib.utils",
|
||||||
|
"codeformer.facelib.utils.misc",
|
||||||
|
"codeformer.facelib.utils.face_restoration_helper",
|
||||||
"compel",
|
"compel",
|
||||||
"controlnet_aux",
|
"controlnet_aux",
|
||||||
"cv2",
|
"cv2",
|
||||||
|
@ -73,6 +78,7 @@ module = [
|
||||||
"safetensors",
|
"safetensors",
|
||||||
"scipy",
|
"scipy",
|
||||||
"timm.models.layers",
|
"timm.models.layers",
|
||||||
|
"torchvision.transforms.functional",
|
||||||
"transformers",
|
"transformers",
|
||||||
"win10toast"
|
"win10toast"
|
||||||
]
|
]
|
||||||
|
|
Loading…
Reference in New Issue