Spaces:
Running
Running
mrolando
commited on
Commit
•
c27c794
1
Parent(s):
1ffabd9
fixed space
Browse files
app.py
CHANGED
@@ -12,7 +12,7 @@ else:
|
|
12 |
device = "cpu"
|
13 |
torch_dtype = torch.float32
|
14 |
print(device)
|
15 |
-
repo_id = "
|
16 |
pipe = AudioLDMPipeline.from_pretrained(repo_id, torch_dtype=torch_dtype)
|
17 |
pipe = pipe.to(device)
|
18 |
|
|
|
12 |
device = "cpu"
|
13 |
torch_dtype = torch.float32
|
14 |
print(device)
|
15 |
+
repo_id = "audioldm-s-full-v2"
|
16 |
pipe = AudioLDMPipeline.from_pretrained(repo_id, torch_dtype=torch_dtype)
|
17 |
pipe = pipe.to(device)
|
18 |
|