fix bug

levo_inference.py

@@ -9,6 +9,7 @@ sys.path.append('.')
 import torch
 
 import json
+import numpy as np
 from omegaconf import OmegaConf
 
 from codeclm.trainer.codec_song_pl import CodecLM_PL
@@ -70,11 +71,12 @@ class LeVoInference(torch.nn.Module):
         params = {**self.default_params, **params}
         self.model.set_generation_params(**params)
 
-        if prompt_audio_path is not None:
+        if prompt_audio_path is not None and os.path.exists(prompt_audio_path):
             pmt_wav, vocal_wav, bgm_wav = self.separator.run(prompt_audio_path)
             melody_is_wav = True
         elif genre is not None and auto_prompt_path is not None:
             auto_prompt = torch.load(auto_prompt_path)
+            merge_prompt = [item for sublist in auto_prompt.values() for item in sublist]
             if genre == "Auto": 
                 prompt_token = merge_prompt[np.random.randint(0, len(merge_prompt))]
             else: