Spaces:
Running
on
Zero
Running
on
Zero
ZeroGPU
Browse files
app.py
CHANGED
|
@@ -30,6 +30,7 @@ device='cuda'
|
|
| 30 |
cfm, tokenizer, muq, vae = prepare_model(device)
|
| 31 |
cfm = torch.compile(cfm)
|
| 32 |
|
|
|
|
| 33 |
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'):
|
| 34 |
|
| 35 |
if randomize_seed:
|
|
|
|
| 30 |
cfm, tokenizer, muq, vae = prepare_model(device)
|
| 31 |
cfm = torch.compile(cfm)
|
| 32 |
|
| 33 |
+
@spaces.GPU(duration=20)
|
| 34 |
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'):
|
| 35 |
|
| 36 |
if randomize_seed:
|