apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix reseed size and mask dims

This commit is contained in:
Sean Sube 2023-11-10 20:00:50 -06:00
parent 8a94cdb385
commit 0d1bf44d25
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/onnx_web/chain/source_txt2img.py
View File

@ -90,7 +90,7 @@ class SourceTxt2ImgStage(BaseStage):
) )
latents[ latents[
:, :, top // 8 : bottom // 8, left // 8 : right // 8 :, :, 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_type = params.get_valid_pipeline("txt2img")
pipe = load_pipeline( pipe = load_pipeline(

4
api/onnx_web/chain/tile.py
View File

@ -106,11 +106,11 @@ def make_tile_mask(
# sort gradient points # sort gradient points
p1_h = adj_tile_h p1_h = adj_tile_h
p2_h = tile_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 p1_w = adj_tile_w
p2_w = tile_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 # build gradients
grad_x, grad_y = [0, 1, 1, 0], [0, 1, 1, 0] grad_x, grad_y = [0, 1, 1, 0], [0, 1, 1, 0]