generate function falls at last hurdle

[markhanslip]

firstly thanks so much for the implementation :)
all runs fine until end of generate function, when this happens:

RuntimeError Traceback (most recent call last)
in
9 first_samples=start_data,
10 #progress_callback=prog_callback, - need to add this to wavenet_model.py
---> 11 temperature=1.)

~/WaveNet/wavenet_model.py in generate(self, num_samples, first_samples, temperature)
230
231 generated = (generated / self.classes) * 2. - 1
--> 232 mu_gen = mu_law_expansion(generated, self.classes)
233
234 self.train()

~/WaveNet/audio_data.py in mu_law_expansion(data, mu)
155
156 def mu_law_expansion(data, mu):
--> 157 s = np.sign(data) * (np.exp(np.abs(data) * np.log(mu + 1)) - 1) / mu
158 return s

RuntimeError: Expected object of type torch.LongTensor but found type torch.DoubleTensor for argument #2 'other'