diff --git a/api/onnx_web/chain/source_txt2img.py b/api/onnx_web/chain/source_txt2img.py index 2442060a..53e31f81 100644 --- 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, _tile_size = dims + pipe_width, pipe_height = size pipe_type = params.get_valid_pipeline("txt2img") pipe = load_pipeline( diff --git a/api/onnx_web/chain/upscale_outpaint.py b/api/onnx_web/chain/upscale_outpaint.py index 0a37d878..dc63fdea 100644 --- a/api/onnx_web/chain/upscale_outpaint.py +++ b/api/onnx_web/chain/upscale_outpaint.py @@ -1,5 +1,5 @@ from logging import getLogger -from typing import Callable, List, Optional +from typing import Callable, List, Optional, Tuple import numpy as np import torch