fix(api): match model path from readme
This commit is contained in:
parent
1f26858aa9
commit
6004f7671c
|
@ -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")
|
Loading…
Reference in New Issue