diff options
Diffstat (limited to 'app/relay/interpreters.js')
| -rw-r--r-- | app/relay/interpreters.js | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/app/relay/interpreters.js b/app/relay/interpreters.js index d892acf..786a5d1 100644 --- a/app/relay/interpreters.js +++ b/app/relay/interpreters.js @@ -23,6 +23,11 @@ export default { params: ['-u'], gpu: true, }, + pytorch_biggan: { + cmd: process.env.PYTORCH_BIGGAN_BIN, + params: ['-u'], + gpu: true, + }, pytorch: { cmd: process.env.PYTORCH_BIN, params: ['-u'], @@ -43,4 +48,4 @@ export default { params: ['-u'], gpu: false, }, -}
\ No newline at end of file +} |
