Spaces:

mrolando
/

text_to_sound

Runtime error

matias commited on Aug 29, 2023

Commit

ee676a1

1 Parent(s): 60d3210

fixed error while processing

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,8 +12,8 @@ else:
 print(device)
 repo_id = "cvssp/audioldm-m-full"
 pipe = AudioLDMPipeline.from_pretrained(repo_id, torch_dtype=torch_dtype)
-pipe = pipe.to(device)
 #pipe.unet = torch.compile(pipe.unet)
 def generate_sound(text):

 print(device)
 repo_id = "cvssp/audioldm-m-full"
 pipe = AudioLDMPipeline.from_pretrained(repo_id, torch_dtype=torch_dtype)
+#pipe = pipe.to(device)
+pipe.unet = torch.compile(pipe.unet)
 #pipe.unet = torch.compile(pipe.unet)
 def generate_sound(text):