apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): seed rng for each image

This commit is contained in:
Sean Sube 2023-01-08 09:31:22 -06:00
parent 282a7cfa2a
commit 8c133e9230
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
api/onnx_web/serve.py
View File

@ -239,12 +239,14 @@ def img2img():
(model, provider, scheduler, prompt, cfg, steps, height,
width, seed, pipe) = pipeline_from_request(OnnxStableDiffusionImg2ImgPipeline)
rng = np.random.RandomState(seed)
image = pipe(
prompt=prompt,
image=input_image,
num_inference_steps=steps,
guidance_scale=cfg,
strength=strength,
generator=rng,
).images[0]
output_file = 'img2img_%s_%s.png' % (seed, spinalcase(prompt[0:64]))
@ -274,6 +276,7 @@ def txt2img():
width, seed, pipe) = pipeline_from_request(OnnxStableDiffusionPipeline)
latents = get_latents_from_seed(seed, width, height)
rng = np.random.RandomState(seed)
image = pipe(
prompt,
@ -281,7 +284,8 @@ def txt2img():
width,
num_inference_steps=steps,
guidance_scale=cfg,
latents=latents
latents=latents,
generator=rng,
).images[0]
output_file = 'txt2img_%s_%s.png' % (seed, spinalcase(prompt[0:64]))