get upscaling model from stage params
This commit is contained in:
parent
18803db6ed
commit
db9189fd3d
|
@ -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)
|
||||||
|
|
|
@ -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))
|
||||||
|
|
|
@ -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
|
||||||
|
|
|
@ -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,
|
||||||
|
|
Loading…
Reference in New Issue