feat(api): add an experimental multi-prompt syntax
This commit is contained in:
parent
2c8ba67ecb
commit
0ad250251e
|
@ -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
|
||||
)
|
||||
|
|
|
@ -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
|
||||
)
|
||||
|
|
Loading…
Reference in New Issue