diff --git a/api/scripts/test-diffusers.py b/api/scripts/test-diffusers.py index 27b018f5..5852eb6b 100644 --- a/api/scripts/test-diffusers.py +++ b/api/scripts/test-diffusers.py @@ -12,9 +12,9 @@ steps = 22 height = 512 width = 512 -model = path.join('..', '..', 'models', 'stable-diffusion-onnx-v1-5') +model = path.join('..', 'models', 'stable-diffusion-onnx-v1-5') prompt = 'an astronaut eating a hamburger' -output = path.join('..', '..', 'outputs', 'test.png') +output = path.join('..', 'outputs', 'test.png') print('generating test image...') pipe = OnnxStableDiffusionPipeline.from_pretrained(model, provider='DmlExecutionProvider', safety_checker=None) diff --git a/api/scripts/test-resrgan.py b/api/scripts/test-resrgan.py index afcbe7ea..efebb80b 100644 --- a/api/scripts/test-resrgan.py +++ b/api/scripts/test-resrgan.py @@ -11,9 +11,9 @@ steps = 22 height = 512 width = 512 -esrgan = path.join('..', '..', 'models', 'RealESRGAN_x4plus.onnx') -output = path.join('..', '..', 'outputs', 'test.png') -upscale = path.join('..', '..', 'outputs', 'test-large.png') +esrgan = path.join('..', 'models', 'RealESRGAN_x4plus.onnx') +output = path.join('..', 'outputs', 'test.png') +upscale = path.join('..', 'outputs', 'test-large.png') print('upscaling test image...') session = ort.InferenceSession(esrgan, providers=['DmlExecutionProvider'])