apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

build full-size latents for outpainting

This commit is contained in:
Sean Sube 2023-01-14 16:28:43 -06:00
parent de13d2d6ef
commit 23ddbc6b10
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/onnx_web/serve.py
View File

@ -268,12 +268,12 @@ def run_inpaint_pipeline(model, provider, scheduler, prompt, negative_prompt, cf
pipe = load_pipeline(OnnxStableDiffusionInpaintPipeline, pipe = load_pipeline(OnnxStableDiffusionInpaintPipeline,
model, provider, scheduler) model, provider, scheduler)
latents = get_latents_from_seed(seed, width, height)
rng = np.random.RandomState(seed) rng = np.random.RandomState(seed)
extra = 256 extra = 256
full_source, full_mask, full_noise, full_dims = expand_image(source_image, mask_image, (extra, extra, extra, extra)) full_source, full_mask, full_noise, full_dims = expand_image(source_image, mask_image, (extra, extra, extra, extra))
full_width, full_height = full_dims full_width, full_height = full_dims
latents = get_latents_from_seed(seed, full_width, full_height)
image = pipe( image = pipe(
prompt, prompt,