apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

use correct generator

This commit is contained in:
Sean Sube 2023-01-28 08:38:48 -06:00
parent 5119a982db
commit 9f76226ba6
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
api/onnx_web/serve.py
View File

@ -24,6 +24,7 @@ from typing import Tuple
from .chain import (
correct_gfpgan,
generate_txt2img,
upscale_outpaint,
upscale_resrgan,
upscale_stable_diffusion,
@ -544,7 +545,7 @@ def chain():
params, size = pipeline_from_request()
example = ChainPipeline(stages=[
(source_txt2img, StageParams(), None),
(generate_txt2img, StageParams(), None),
(upscale_outpaint, StageParams(outscale=4), {
'expand': Border(256, 256, 256, 256),
}),