app/relay/modules/biggan.js


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25

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 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: 'cli/cli.py',
  params: (task) => {
    console.log(task)
    const opt = task.opt || {}
    return ["biggan", "live"]
  }
}

export default {
  name, cwd, env,
  activities: {
    live,
  }
}