fix(scripts): correct model paths in test scripts

ssube · Apr 17, 2023 · e12f3c2 · e12f3c2
1 parent 64ea060
commit e12f3c2
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/api/scripts/test-diffusers.py b/api/scripts/test-diffusers.py
@@ -12,9 +12,9 @@
 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
@@ -11,9 +11,9 @@
 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'])