Change default device to cuda
Browse files
app.py
CHANGED
@@ -10,7 +10,7 @@ from audioldm.variational_autoencoder import AutoencoderKL
|
|
10 |
from gradio import Markdown
|
11 |
|
12 |
class Tango:
|
13 |
-
def __init__(self, name="declare-lab/tango", device="
|
14 |
|
15 |
path = snapshot_download(repo_id=name)
|
16 |
|
|
|
10 |
from gradio import Markdown
|
11 |
|
12 |
class Tango:
|
13 |
+
def __init__(self, name="declare-lab/tango", device="cuda:0"):
|
14 |
|
15 |
path = snapshot_download(repo_id=name)
|
16 |
|