ensure tiled latents are correct shape

ssube · Jul 10, 2023 · 8acfaaa · 8acfaaa
1 parent da6f00d
commit 8acfaaa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/api/onnx_web/chain/source_txt2img.py b/api/onnx_web/chain/source_txt2img.py
@@ -65,7 +65,7 @@ def run(
         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(