ensure tiled latents are correct shape

2023-07-09 22:45:51 -05:00 · 2023-07-09 22:45:51 -05:00 · 8acfaaa1ee
parent da6f00d2a5
commit 8acfaaa1ee
1 changed files with 1 additions and 1 deletions
--- a/api/onnx_web/chain/source_txt2img.py
+++ b/api/onnx_web/chain/source_txt2img.py
@ -65,7 +65,7 @@ class SourceTxt2ImgStage(BaseStage):
        else:
            # slice existing latents
            latents = get_tile_latents(latents, dims, size)
-            pipe_width, pipe_height = size
+            pipe_width = pipe_height = tile_size

        pipe_type = params.get_valid_pipeline("txt2img")
        pipe = load_pipeline(