apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

get upscaling model from stage params

This commit is contained in:
Sean Sube 2023-01-28 22:52:39 -06:00
parent 18803db6ed
commit db9189fd3d
4 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/onnx_web/chain/blend_img2img.py
View File

@ -30,7 +30,7 @@ def blend_img2img(
prompt: str = None, prompt: str = None,
**kwargs, **kwargs,
) -> Image.Image: ) -> Image.Image:
logger.info('generating image using img2img', params.prompt) logger.info('generating image using img2img, %s steps: %s', params.steps, params.prompt)
pipe = load_pipeline(OnnxStableDiffusionImg2ImgPipeline, pipe = load_pipeline(OnnxStableDiffusionImg2ImgPipeline,
params.model, params.provider, params.scheduler) params.model, params.provider, params.scheduler)

2
api/onnx_web/serve.py
View File

@ -588,7 +588,7 @@ def chain():
# TODO: create Upscale from upscale # TODO: create Upscale from upscale
if 'upscale' in kwargs: if 'upscale' in kwargs:
upscale = UpscaleParams(params.model, params.provider) upscale = UpscaleParams(kwargs.get('upscale'), params.provider)
kwargs['upscale'] = upscale kwargs['upscale'] = upscale
pipeline.append((callback, stage, kwargs)) pipeline.append((callback, stage, kwargs))

2
api/schema.yaml
View File

@ -16,14 +16,12 @@ $defs:
image_params: image_params:
type: object type: object
required: [prompt]
properties: properties:
prompt: prompt:
type: string type: string
upscale_params: upscale_params:
type: object type: object
required: [outscale, scale]
properties: properties:
outscale: outscale:
type: number type: number

2
common/pipelines/example.json
View File

@ -19,7 +19,7 @@
"name": "upscale", "name": "upscale",
"type": "upscale-stable-diffusion", "type": "upscale-stable-diffusion",
"params": { "params": {
"model": "stable-diffusion-x4-upscaler", "upscale": "stable-diffusion-x4-upscaler",
"prompt": "a magical wizard in a robe fighting a dragon", "prompt": "a magical wizard in a robe fighting a dragon",
"scale": 4, "scale": 4,
"outscale": 4, "outscale": 4,