diff options
| -rw-r--r-- | app/relay/modules/biggan.js | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/app/relay/modules/biggan.js b/app/relay/modules/biggan.js index 6d6e211..44e8e4e 100644 --- a/app/relay/modules/biggan.js +++ b/app/relay/modules/biggan.js @@ -2,7 +2,7 @@ import path from 'path' import fs from 'fs' const name = 'biggan' -const cwd = process.env.BIGGAN_CWD || path.join(process.env.HOME, 'code/' + name + '/') +const cwd = process.env.BIGGAN_CWD || path.join(process.env.HOME, 'code/' + name + '/cli/') const env = { LD_LIBRARY_PATH: '/usr/local/cuda/lib64:' + process.env.TENSORRT_LIB_PATH, TFHUB_CACHE_DIR: '/home/lens/.tfhub', @@ -11,7 +11,7 @@ const env = { } const live = { type: 'pytorch_biggan', - script: 'cli/cli.py', + script: 'cli.py', params: (task) => { console.log(task) const opt = task.opt || {} |