apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

resize tiles before running refinement steps

This commit is contained in:
Sean Sube 2023-03-31 23:39:25 -05:00
parent 66e938f9ab
commit 4ab1b6cc88
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 13 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
api/onnx_web/diffusers/run.py
View File

@ -87,7 +87,7 @@ def run_txt2img_pipeline(
for image, output in zip(result.images, outputs):
if highres_scale > 1:
def highres(tile: Image.Image, dims):
# load img2img pipeline once
highpipe = load_pipeline(
server,
OnnxStableDiffusionImg2ImgPipeline,
@ -98,7 +98,9 @@ def run_txt2img_pipeline(
inversions,
loras,
)
progress = job.get_progress_callback()
def highres(tile: Image.Image, dims):
tile = tile.resize((size.height, size.width))
if params.lpw:
logger.debug("using LPW pipeline for img2img")
rng = torch.manual_seed(params.seed)