img opz

2 files changed, 6 insertions, 2 deletions

diff --git a/app/relay/modules/pix2pixhd.js b/app/relay/modules/pix2pixhd.js
index 5b9fd5c..34fc972 100644
--- a/app/relay/modules/pix2pixhd.js
+++ b/app/relay/modules/pix2pixhd.js
@@ -3,6 +3,10 @@ import fs from 'fs'
 
 const name = 'pix2pixhd'
 const cwd = process.env.PIX2PIXHD_CWD || path.join(process.env.HOME, 'code/' + name + '/')
+const env = {
+  LD_LIBRARY_PATH: process.env.HOME + '/Downloads/TensorRT-3.0.4/lib',
+}
+console.log(env)
 
 const fetch = {
   type: 'perl',
@@ -101,7 +105,6 @@ const live = {
     console.log(task)
     const opt = task.opt || {}
     return [
-      'LD_LIBRARY_PATH=' + process.env.HOME + '/Downloads/TensorRT-3.0.4/lib',
       '--dataroot', path.join(cwd, 'sequences', task.dataset),
       '--start_img', path.join(cwd, 'sequences', task.dataset, 'frame_00001.png'),
       '--checkpoint-name', task.checkpoint,
@@ -128,7 +131,7 @@ const live = {
 }
 
 export default {
-  name, cwd,
+  name, cwd, env,
   activities: {
     fetch,
     build,
diff --git a/app/relay/runner.js b/app/relay/runner.js
index 6464288..531bf12 100644
--- a/app/relay/runner.js
+++ b/app/relay/runner.js
@@ -261,6 +261,7 @@ export function run_task_with_activity(task, module, activity, preempt=false, wa
   task.processing = true
 	const subprocess = spawn(interpreter.cmd, params, {
     cwd: module.cwd,
+    env: module.env || {},
   })
 	if (interpreter.gpu) {
 		state.current_gpu_task = {