From 6004f7671c011d291268264416b4824ef9692107 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Thu, 5 Jan 2023 15:38:29 -0600 Subject: [PATCH] fix(api): match model path from readme --- api/test-setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/api/test-setup.py b/api/test-setup.py index fa01feb1..c3955d14 100644 --- a/api/test-setup.py +++ b/api/test-setup.py @@ -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") \ No newline at end of file