Update inference/mastering_transfer.py

inference/mastering_transfer.py

@@ -89,6 +89,8 @@ class Mastering_Style_Transfer_Inference:
 
         input_aud = load_wav_segment(input_track_path, axis=0)
         reference_aud = load_wav_segment(reference_track_path, axis=0)
+        input_aud = torch.FloatTensor(input_aud).to(self.device)
+        reference_aud = torch.FloatTensor(reference_aud).to(self.device)
 
         cur_out_dir = './yt_dir/0/'
         os.makedirs(cur_out_dir, exist_ok=True)