fix bug

levo_inference.py

@@ -107,6 +107,9 @@ class LeVoInference(torch.nn.Module):
             tokens = tokens[..., :3000]
             
         with torch.no_grad():
-            wav_seperate = self.model.generate_audio(tokens, pmt_wav, vocal_wav, bgm_wav)
+            if melody_is_wav:
+                wav_seperate = self.model.generate_audio(tokens, pmt_wav, vocal_wav, bgm_wav)
+            else:
+                wav_seperate = self.model.generate_audio(tokens)
 
         return wav_seperate[0]