1 files changed, 7 insertions, 3 deletions
diff --git a/app/client/modules/pix2wav/views/pix2wav.live.js b/app/client/modules/pix2wav/views/pix2wav.live.js
index 2902bff..9a61e60 100644
--- a/app/client/modules/pix2wav/views/pix2wav.live.js
+++ b/app/client/modules/pix2wav/views/pix2wav.live.js
@@ -28,6 +28,7 @@ class Pix2WavLive extends Component {
   }
   componentWillUpdate(nextProps) {
     if (nextProps.opt.checkpoint_name && nextProps.opt.checkpoint_name !== this.props.opt.checkpoint_name) {
+      console.log('fetch checkpoint', nextProps.opt.checkpoint_name)
       this.props.actions.live.list_epochs('pix2wav', nextProps.opt.checkpoint_name)
     }
   }
@@ -47,8 +48,9 @@ class Pix2WavLive extends Component {
   }
   start(){
     // console.log(this.props)
-    const sequence = this.props.pix2wav.data.sequences[0].name
-    const checkpoint = this.props.pix2wav.data.checkpoints[0].name
+    console.log(this.props.pix2wav.data)
+    const sequence = this.props.pix2wav.data.sequences[0].name || ''
+    const checkpoint = this.props.pix2wav.data.checkpoints[0].name || ''
     console.log('starting up!', sequence, checkpoint)
     this.props.actions.tasks.live_task(sequence, checkpoint)
   }
@@ -75,6 +77,8 @@ class Pix2WavLive extends Component {
     if (this.props.pix2wav.loading) {
       return <Loading />
     }
+    // console.log('sequence', this.props.opt)
+    console.log(this.props.pix2wav.data.checkpoints)
     return (
       <div className='app centered'>
         <Player width={424} height={256} />
@@ -255,7 +259,7 @@ class Pix2WavLive extends Component {
         >Start</Button>
       )
     }
-    if (this.props.runner.gpu.task.module !== 'pix2wav') {
+    if (this.props.runner.gpu.task.module !== 'pix2pix' && this.props.runner.gpu.task.module !== 'pix2wav') {
       return (
         <Button
           title={'GPU Busy'}