ZeroGPU

app.py

@@ -30,6 +30,7 @@ device='cuda'
 cfm, tokenizer, muq, vae = prepare_model(device)
 cfm = torch.compile(cfm)
 
+@spaces.GPU(duration=20)
 def infer_music(lrc, ref_audio_path, text_prompt, current_prompt_type, seed=42, randomize_seed=False, steps=32, file_type='wav', max_frames=2048, device='cuda'):
 
     if randomize_seed: