full model

app.py

@@ -31,7 +31,7 @@ cfm, cfm_full, tokenizer, muq, vae = prepare_model(device)
 cfm = torch.compile(cfm)
 cfm_full = torch.compile(cfm_full)
 
-@spaces.GPU(duration=20)
+@spaces.GPU(duration=40)
 def infer_music(lrc, ref_audio_path, text_prompt, current_prompt_type, seed=42, randomize_seed=False, steps=32, cfg_strength=4.0, file_type='wav', odeint_method='euler', Music_Duration='95s', device='cuda'):
     if Music_Duration == '95s':
         max_frames = 2048
@@ -45,7 +45,7 @@ def infer_music(lrc, ref_audio_path, text_prompt, current_prompt_type, seed=42,
     sway_sampling_coef = -1 if steps < 32 else None
     vocal_flag = False
     try:
-        lrc_prompt, start_time = get_lrc_token(lrc, tokenizer, device)
+        lrc_prompt, start_time = get_lrc_token(max_frames, lrc, tokenizer, device)
         if current_prompt_type == 'audio':
             style_prompt, vocal_flag = get_audio_style_prompt(muq, ref_audio_path)
         else:
@@ -415,4 +415,4 @@ with gr.Blocks(css=css) as demo:
 
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(share=True)