Spaces:

mrolando
/

text_to_sound

Runtime error

mrolando commited on Oct 18, 2023

Commit

44c548d

•

1 Parent(s): 0592ce5

changed model

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,7 +14,7 @@ else:
     device = "cpu"
     torch_dtype = torch.float32
 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)

     device = "cpu"
     torch_dtype = torch.float32
 print(device)
+repo_id = "cvssp/audioldm-s-full-v2"
 pipe = AudioLDMPipeline.from_pretrained(repo_id, torch_dtype=torch_dtype)
 pipe = pipe.to(device)
 # pipe.unet = torch.compile(pipe.unet)