update

app.py

@@ -30,7 +30,7 @@ cfm, tokenizer, muq, vae = prepare_model(device)
 #cfm = torch.compile(cfm)
 
 #@spaces.GPU
-def infer_music(lrc, ref_audio_path, steps, file_type, max_frames=2048, device='cuda'):
+def infer_music(lrc, ref_audio_path, steps, file_type, max_frames=2048):
 
     sway_sampling_coef = -1 if steps < 32 else None
     lrc_prompt, start_time = get_lrc_token(lrc, tokenizer, device)
@@ -327,7 +327,7 @@ with gr.Blocks(css=css) as demo:
     
     lyrics_btn.click(
         fn=infer_music,
-        inputs=[lrc, audio_prompt, steps, file_type, device],
+        inputs=[lrc, audio_prompt, steps, file_type],
         outputs=audio_output
     )