CPU

app.py

@@ -11,7 +11,7 @@ from gradio import Markdown
 import spaces
 
 class Tango:
-    def __init__(self, name="declare-lab/tango2", device="cuda:0"):
+    def __init__(self, name="declare-lab/tango2", device="cpu"):
         
         path = snapshot_download(repo_id=name)
         
@@ -70,9 +70,9 @@ class Tango:
 # Initialize TANGO
 
 tango = Tango(device="cpu")
-tango.vae.to("cuda")
-tango.stft.to("cuda")
-tango.model.to("cuda")
+tango.vae.to("cpu")
+tango.stft.to("cpu")
+tango.model.to("cpu")
     
 @spaces.GPU(duration=60)
 def gradio_generate(prompt, steps, guidance):