apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

feat(api): add an experimental multi-prompt syntax

This commit is contained in:
Sean Sube 2023-08-25 23:31:43 -05:00
parent 2c8ba67ecb
commit 0ad250251e
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 13 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -6,7 +6,7 @@ import torch
from PIL import Image from PIL import Image
from ..diffusers.load import load_pipeline from ..diffusers.load import load_pipeline
from ..diffusers.utils import encode_prompt, parse_prompt from ..diffusers.utils import encode_prompt, parse_prompt, slice_prompt
from ..params import ImageParams, SizeChart, StageParams from ..params import ImageParams, SizeChart, StageParams
from ..server import ServerContext from ..server import ServerContext
from ..worker import ProgressCallback, WorkerContext from ..worker import ProgressCallback, WorkerContext
@ -32,6 +32,12 @@ class BlendImg2ImgStage(BaseStage):
**kwargs, **kwargs,
) -> List[Image.Image]: ) -> List[Image.Image]:
params = params.with_args(**kwargs) params = params.with_args(**kwargs)
# highres hax
params = params.with_args(
prompt = slice_prompt(params.prompt, 1)
)
logger.info( logger.info(
"blending image using img2img, %s steps: %s", params.steps, params.prompt "blending image using img2img, %s steps: %s", params.steps, params.prompt
) )

6
api/onnx_web/chain/source_txt2img.py
View File

@ -11,6 +11,7 @@ from ..diffusers.utils import (
get_latents_from_seed, get_latents_from_seed,
get_tile_latents, get_tile_latents,
parse_prompt, parse_prompt,
slice_prompt,
) )
from ..params import ImageParams, Size, SizeChart, StageParams from ..params import ImageParams, Size, SizeChart, StageParams
from ..server import ServerContext from ..server import ServerContext
@ -40,6 +41,11 @@ class SourceTxt2ImgStage(BaseStage):
params = params.with_args(**kwargs) params = params.with_args(**kwargs)
size = size.with_args(**kwargs) size = size.with_args(**kwargs)
# highres hax
params = params.with_args(
prompt = slice_prompt(params.prompt, 0)
)
logger.info( logger.info(
"generating image using txt2img, %s steps: %s", params.steps, params.prompt "generating image using txt2img, %s steps: %s", params.steps, params.prompt
) )