apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): match model path from readme

This commit is contained in:
Sean Sube 2023-01-05 15:38:29 -06:00
parent 1f26858aa9
commit 6004f7671c
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/test-setup.py
View File

@ -6,6 +6,6 @@ height = 512
width = 512
prompt = "an astronaut eating a hamburger"
pipe = OnnxStableDiffusionPipeline.from_pretrained("../../stable_diffusion_onnx", provider="DmlExecutionProvider", safety_checker=None)
pipe = OnnxStableDiffusionPipeline.from_pretrained("../models/stable-diffusion-onnx-v1-5", provider="DmlExecutionProvider", safety_checker=None)
image = pipe(prompt, height, width, num_inference_steps=steps, guidance_scale=cfg).images[0]
image.save("../test.png")