fix bug

codeclm/tokenizer/Flow1dVAE/generate_septoken.py

@@ -188,7 +188,7 @@ class Tango:
         first_latent_codes_length = 0
 
 
-        if(isinstance(prompt_vocal, torch.Tensor)):
+        if (isinstance(prompt_vocal, torch.Tensor)) and (isinstance(prompt_bgm, torch.Tensor)):
             # prepare prompt
             prompt_vocal = prompt_vocal.to(self.device)
             prompt_bgm = prompt_bgm.to(self.device)

tools/gradio/levo_inference_lowmem.py

@@ -115,9 +115,9 @@ class LeVoInference(torch.nn.Module):
             
         with torch.no_grad():
             if melody_is_wav:
-                wav_seperate = self.model.generate_audio(tokens, pmt_wav, vocal_wav, bgm_wav, gen_type=gen_type)
+                wav_seperate = model.generate_audio(tokens, pmt_wav, vocal_wav, bgm_wav, gen_type=gen_type)
             else:
-                wav_seperate = self.model.generate_audio(tokens, gen_type=gen_type)
+                wav_seperate = model.generate_audio(tokens, gen_type=gen_type)
 
         del seperate_tokenizer
         del model