Improved version compatibility

model.py

@@ -105,7 +105,7 @@ def decode(self, labels=None, condition=None, epsilon=None):
                 eps = [None] * self.hps.num_levels
 
             z = sample(eps=eps[-1])
-            objective = tf.zeros(tf.shape(z)[0])
+            objective = tf.zeros(self.hps.batch_size)
 
         if self.hps.conditioning and condition is None:
             condition = y_onehot