diff --git a/api/onnx_web/chain/blend_img2img.py b/api/onnx_web/chain/blend_img2img.py index 34986d52..aca68558 100644 --- a/api/onnx_web/chain/blend_img2img.py +++ b/api/onnx_web/chain/blend_img2img.py @@ -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 ) diff --git a/api/onnx_web/chain/source_txt2img.py b/api/onnx_web/chain/source_txt2img.py index 677f1d6a..3120d1ec 100644 --- a/api/onnx_web/chain/source_txt2img.py +++ b/api/onnx_web/chain/source_txt2img.py @@ -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 )