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,
|
||||
**kwargs,
|
||||
) -> 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,
|
||||
params.model, params.provider, params.scheduler)
|
||||
|
|
|
@ -588,7 +588,7 @@ def chain():
|
|||
|
||||
# TODO: create Upscale from upscale
|
||||
if 'upscale' in kwargs:
|
||||
upscale = UpscaleParams(params.model, params.provider)
|
||||
upscale = UpscaleParams(kwargs.get('upscale'), params.provider)
|
||||
kwargs['upscale'] = upscale
|
||||
|
||||
pipeline.append((callback, stage, kwargs))
|
||||
|
|
|
@ -16,14 +16,12 @@ $defs:
|
|||
|
||||
image_params:
|
||||
type: object
|
||||
required: [prompt]
|
||||
properties:
|
||||
prompt:
|
||||
type: string
|
||||
|
||||
upscale_params:
|
||||
type: object
|
||||
required: [outscale, scale]
|
||||
properties:
|
||||
outscale:
|
||||
type: number
|
||||
|
|
|
@ -19,7 +19,7 @@
|
|||
"name": "upscale",
|
||||
"type": "upscale-stable-diffusion",
|
||||
"params": {
|
||||
"model": "stable-diffusion-x4-upscaler",
|
||||
"upscale": "stable-diffusion-x4-upscaler",
|
||||
"prompt": "a magical wizard in a robe fighting a dragon",
|
||||
"scale": 4,
|
||||
"outscale": 4,
|
||||
|
|
Loading…
Reference in New Issue