Spaces:

ASLP-lab
/

DiffRhythm2

Running on Zero

ASLP-lab commited on 22 days ago

Commit

2e9828f

verified ·

1 Parent(s): 183fdab

add space gpu

Files changed (1) hide show

app.py CHANGED Viewed

@@ -224,8 +224,8 @@ device='cuda'
 dtype=torch.float16
 diffrhythm2, mulan, lrc_tokenizer, decoder = prepare_model("ASLP-Lab/DiffRhythm2", device, dtype)
-# import spaces
-# @spaces.GPU
 def infer_music(
         lrc,
         current_prompt_type,

 dtype=torch.float16
 diffrhythm2, mulan, lrc_tokenizer, decoder = prepare_model("ASLP-Lab/DiffRhythm2", device, dtype)
+import spaces
+@spaces.GPU
 def infer_music(
         lrc,
         current_prompt_type,