diff options
Diffstat (limited to 'app/relay/interpreters.js')
| -rw-r--r-- | app/relay/interpreters.js | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/app/relay/interpreters.js b/app/relay/interpreters.js index b1507e6..d892acf 100644 --- a/app/relay/interpreters.js +++ b/app/relay/interpreters.js @@ -18,6 +18,11 @@ export default { params: ['-u'], gpu: false, }, + pytorch_samplernn: { + cmd: process.env.PYTORCH_SAMPLERNN_BIN, + params: ['-u'], + gpu: true, + }, pytorch: { cmd: process.env.PYTORCH_BIN, params: ['-u'], |