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
|
||||
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)
|
||||
|
|
|
@ -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'])
|
||||
|
|
Loading…
Reference in New Issue