diff --git a/api/onnx_web/chain/source_txt2img.py b/api/onnx_web/chain/source_txt2img.py index 8a4dd61b..cf2f039d 100644 --- a/api/onnx_web/chain/source_txt2img.py +++ b/api/onnx_web/chain/source_txt2img.py @@ -90,7 +90,7 @@ class SourceTxt2ImgStage(BaseStage): ) latents[ :, :, top // 8 : bottom // 8, left // 8 : right // 8 - ] = get_latents_from_seed(region_seed, latent_size, params.batch) + ] = get_latents_from_seed(region_seed, Size(right - left, bottom - top), params.batch) pipe_type = params.get_valid_pipeline("txt2img") pipe = load_pipeline( diff --git a/api/onnx_web/chain/tile.py b/api/onnx_web/chain/tile.py index 4d8bee2d..e3505984 100644 --- a/api/onnx_web/chain/tile.py +++ b/api/onnx_web/chain/tile.py @@ -106,11 +106,11 @@ def make_tile_mask( # sort gradient points p1_h = adj_tile_h p2_h = tile_h - adj_tile_h - points_h = [0, min(p1_h, p2_h), max(p1_h, p2_h), tile] + points_h = [0, min(p1_h, p2_h), max(p1_h, p2_h), tile_h] p1_w = adj_tile_w p2_w = tile_w - adj_tile_w - points_w = [0, min(p1_w, p2_w), max(p1_w, p2_w), tile] + points_w = [0, min(p1_w, p2_w), max(p1_w, p2_w), tile_w] # build gradients grad_x, grad_y = [0, 1, 1, 0], [0, 1, 1, 0]