fix(scripts): correct model paths in test scripts
This commit is contained in:
parent
64ea060ec9
commit
e12f3c2801
|
@ -12,9 +12,9 @@ steps = 22
|
||||||
height = 512
|
height = 512
|
||||||
width = 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'
|
prompt = 'an astronaut eating a hamburger'
|
||||||
output = path.join('..', '..', 'outputs', 'test.png')
|
output = path.join('..', 'outputs', 'test.png')
|
||||||
|
|
||||||
print('generating test image...')
|
print('generating test image...')
|
||||||
pipe = OnnxStableDiffusionPipeline.from_pretrained(model, provider='DmlExecutionProvider', safety_checker=None)
|
pipe = OnnxStableDiffusionPipeline.from_pretrained(model, provider='DmlExecutionProvider', safety_checker=None)
|
||||||
|
|
|
@ -11,9 +11,9 @@ steps = 22
|
||||||
height = 512
|
height = 512
|
||||||
width = 512
|
width = 512
|
||||||
|
|
||||||
esrgan = path.join('..', '..', 'models', 'RealESRGAN_x4plus.onnx')
|
esrgan = path.join('..', 'models', 'RealESRGAN_x4plus.onnx')
|
||||||
output = path.join('..', '..', 'outputs', 'test.png')
|
output = path.join('..', 'outputs', 'test.png')
|
||||||
upscale = path.join('..', '..', 'outputs', 'test-large.png')
|
upscale = path.join('..', 'outputs', 'test-large.png')
|
||||||
|
|
||||||
print('upscaling test image...')
|
print('upscaling test image...')
|
||||||
session = ort.InferenceSession(esrgan, providers=['DmlExecutionProvider'])
|
session = ort.InferenceSession(esrgan, providers=['DmlExecutionProvider'])
|
||||||
|
|
Loading…
Reference in New Issue