Arthur Islamov
|
c262879214
|
Added DML and CUDA provider support in onnxruntime-node (#16050)
### Description
I've added changes to support CUDA and DML (only on Windows, on other
platforms it will throw an error)
### Motivation and Context
It fixes this feature request
https://github.com/microsoft/onnxruntime/issues/14127 which is tracked
here https://github.com/microsoft/onnxruntime/issues/14529
I was working on StableDiffusion implementation for node.js and it is
very slow on CPU, so GPU support is essential.
Here is a working demo with a patched and precompiled version
https://github.com/dakenf/stable-diffusion-nodejs
---------
|
2023-08-25 16:57:06 -07:00 |
|