Spaces:

inoculatemedia
/

AudioLDM

Runtime error

inoculatemedia commited on 14 days ago

Commit

4eefb19

verified ·

1 Parent(s): df7d046

Update audioldm/pipeline.py

Files changed (1) hide show

audioldm/pipeline.py CHANGED Viewed

@@ -161,7 +161,7 @@ def style_transfer(
         device = torch.device("cuda:0")
     else:
         device = torch.device("cpu")
     assert original_audio_file_path is not None, "You need to provide the original audio file path"
     audio_file_duration = get_duration(original_audio_file_path)

         device = torch.device("cuda:0")
     else:
         device = torch.device("cpu")
+    @spaces.GPU
     assert original_audio_file_path is not None, "You need to provide the original audio file path"
     audio_file_duration = get_duration(original_audio_file_path)