1
0
Fork 0

apply lint

This commit is contained in:
Sean Sube 2023-06-08 07:20:19 -05:00
parent 3f00da9681
commit f4d944acbe
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 10 additions and 3 deletions

View File

@ -71,7 +71,9 @@ def blend_tiles(
overlap: float,
):
adj_tile = int(float(tile) * (1.0 - overlap))
logger.trace("adjusting tile size from %s to %s based on %s overlap", tile, adj_tile, overlap)
logger.trace(
"adjusting tile size from %s to %s based on %s overlap", tile, adj_tile, overlap
)
scaled_size = (height * scale, width * scale, 3)
count = np.zeros(scaled_size)
@ -100,7 +102,13 @@ def blend_tiles(
# equalized size may be wrong/too much
scaled_bottom = min(scaled_top + equalized.shape[0], scaled_size[0])
scaled_right = min(scaled_left + equalized.shape[1], scaled_size[1])
logger.trace("tile broadcast shapes: %s, %s, %s, %s", scaled_top, scaled_bottom, scaled_left, scaled_right)
logger.trace(
"tile broadcast shapes: %s, %s, %s, %s",
scaled_top,
scaled_bottom,
scaled_left,
scaled_right,
)
# accumulation
value[scaled_top:scaled_bottom, scaled_left:scaled_right, :] += equalized[

View File

@ -46,7 +46,6 @@ class ConversionContext(ServerContext):
reload: bool = True,
share_unet: bool = True,
extract: bool = False,
**kwargs,
) -> None:
super().__init__(model_path=model_path, cache_path=cache_path, **kwargs)