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 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
|
||||||
)
|
)
|
||||||
|
|
|
@ -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
|
||||||
)
|
)
|
||||||
|
|
Loading…
Reference in New Issue