From 76bed9582b2cc9febbdd1938716c6ef0206a8de9 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sun, 9 Jul 2023 22:28:23 -0500 Subject: [PATCH] fixes for txt2img --- api/onnx_web/chain/source_txt2img.py | 2 +- api/onnx_web/chain/upscale_outpaint.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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