summaryrefslogtreecommitdiff
path: root/app/relay/interpreters.js
diff options
context:
space:
mode:
Diffstat (limited to 'app/relay/interpreters.js')
-rw-r--r--app/relay/interpreters.js7
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
+}