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 ..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 ..server import ServerContext
from ..worker import ProgressCallback, WorkerContext
@ -32,6 +32,12 @@ class BlendImg2ImgStage(BaseStage):
**kwargs,
) -> List[Image.Image]:
params = params.with_args(**kwargs)
# highres hax
params = params.with_args(
prompt = slice_prompt(params.prompt, 1)
)
logger.info(
"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_tile_latents,
parse_prompt,
slice_prompt,
)
from ..params import ImageParams, Size, SizeChart, StageParams
from ..server import ServerContext
@ -40,6 +41,11 @@ class SourceTxt2ImgStage(BaseStage):
params = params.with_args(**kwargs)
size = size.with_args(**kwargs)
# highres hax
params = params.with_args(
prompt = slice_prompt(params.prompt, 0)
)
logger.info(
"generating image using txt2img, %s steps: %s", params.steps, params.prompt
)