apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(scripts): correct model paths in test scripts

This commit is contained in:
Sean Sube 2023-04-16 23:18:35 -05:00
parent 64ea060ec9
commit e12f3c2801
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/scripts/test-diffusers.py
View File

@ -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)

6
api/scripts/test-resrgan.py
View File

@ -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'])