diff options
Diffstat (limited to 'app/relay/modules')
| -rw-r--r-- | app/relay/modules/biggan.js | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/app/relay/modules/biggan.js b/app/relay/modules/biggan.js index 97042be..9dfa6f7 100644 --- a/app/relay/modules/biggan.js +++ b/app/relay/modules/biggan.js @@ -3,6 +3,10 @@ import fs from 'fs' const name = 'biggan' const cwd = process.env.BIGGAN_CWD || path.join(process.env.HOME, 'code/' + name + '/') +const env = { + LD_LIBRARY_PATH: '/usr/local/cuda/lib64:' + process.env.TENSORRT_LIB_PATH, + TFHUB_CACHE_DIR: '/home/lens/.tfhub', +} const live = { type: 'pytorch_biggan', script: 'inversion/live.py', @@ -14,7 +18,7 @@ const live = { } export default { - name, cwd, + name, cwd, env, activities: { live, } |