update

app.py

@@ -22,13 +22,14 @@ from diffrhythm.infer.infer_utils import (
     get_negative_style_prompt
 )
 from diffrhythm.infer.infer import inference
+import devicetorch
 
 
-device='cuda'
+device=devicetorch.get(torch)
 cfm, tokenizer, muq, vae = prepare_model(device)
 cfm = torch.compile(cfm)
 
-@spaces.GPU
+#@spaces.GPU
 def infer_music(lrc, ref_audio_path, steps, file_type, max_frames=2048, device='cuda'):
 
     sway_sampling_coef = -1 if steps < 32 else None
@@ -326,7 +327,7 @@ with gr.Blocks(css=css) as demo:
     
     lyrics_btn.click(
         fn=infer_music,
-        inputs=[lrc, audio_prompt, steps, file_type],
+        inputs=[lrc, audio_prompt, steps, file_type, device],
         outputs=audio_output
     )
 
@@ -336,4 +337,4 @@ demo.queue().launch(show_api=False, show_error=True)
 
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()

requirements.txt

@@ -1,8 +1,8 @@
 gradio==5.20.0
 accelerate==1.4.0
 inflect==7.5.0
-torchdiffeq==0.2.5
-torchaudio==2.6.0
+#torchdiffeq==0.2.5
+#torchaudio==2.6.0
 x-transformers==2.1.2
 transformers==4.49.0
 librosa==0.10.2.post1
@@ -30,4 +30,4 @@ einops==0.8.1
 lazy_loader==0.4
 scipy==1.15.2
 ftfy==6.3.1
-torchdiffeq==0.2.5
+torchdiffeq==0.2.5