fix reseed size and mask dims
This commit is contained in:
parent
8a94cdb385
commit
0d1bf44d25
|
@ -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(
|
||||||
|
|
|
@ -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]
|
||||||
|
|
Loading…
Reference in New Issue