diff options
| -rw-r--r-- | app/relay/modules/pix2pixhd.js | 21 |
1 files changed, 18 insertions, 3 deletions
diff --git a/app/relay/modules/pix2pixhd.js b/app/relay/modules/pix2pixhd.js index b4a206a..c09eadc 100644 --- a/app/relay/modules/pix2pixhd.js +++ b/app/relay/modules/pix2pixhd.js @@ -137,6 +137,21 @@ const augment = { } return args }, + listen: (task, res, i) => { + // extract the new path name + const lines = res.split('\n') + for (let line of lines) { + console.log(line) + if ( line.match(/^render_dir: /) ) { + let tag = line.split(': ')[1].trim() + task.opt.render_dir = tag + // task.opt.filename = filename + console.log(">>>>>> created dataset", tag) + return { type: 'progress', action: 'resolve_dataset', task } + } + } + return null + }, after: 'render_recursive', } const clear_recursive = { @@ -210,11 +225,11 @@ const render_recursive = { console.log('will not render mov') return 'CANCEL' } - const tag = task.dataset + '/' + pad(task.opt.epoch || 'latest', 2) + '_' + pad(task.opt.augment_take, 5) + '_' + pad(task.opt.augment_make, 5) - console.log('rendering recursve path:', tag) + const render_dir = task.opt.render_dir.replace('./recursive/', '') + console.log('rendering recursve path:', render_dir) return [ '--path', 'recursive', - '--tag', tag, + '--tag', render_dir, '--module', task.module, '--endpoint', process.env.API_REMOTE + '/api/folder/' + task.opt.folder_id + '/upload/', ] |