From 0d1bf44d2563cdd9ef690153c1da89c9fa4979a2 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Fri, 10 Nov 2023 20:00:50 -0600 Subject: [PATCH] fix reseed size and mask dims --- api/onnx_web/chain/source_txt2img.py | 2 +- api/onnx_web/chain/tile.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) 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]