From e12f3c2801122ac383d2a12b15b086e5a14e74fd Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sun, 16 Apr 2023 23:18:35 -0500 Subject: [PATCH] fix(scripts): correct model paths in test scripts --- api/scripts/test-diffusers.py | 4 ++-- api/scripts/test-resrgan.py | 6 +++--- 2 files changed, 5 insertions(+), 5 deletions(-) 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'])