Spaces:

declare-lab
/

tango2

Running on Zero

hungchiayu1 commited on Apr 14

Commit

a258601

•

1 Parent(s): 88895ae

fixed bugs

Files changed (1) hide show

app.py CHANGED Viewed

@@ -68,10 +68,9 @@ class Tango:
             return list(self.chunks(outputs, samples))
 # Initialize TANGO
-if torch.cuda.is_available():
-    tango = Tango(device="cuda")
-else:
-    tango = Tango(device="cpu")
 @spaces.GPU(duration=240)
 def gradio_generate(prompt, steps, guidance):

             return list(self.chunks(outputs, samples))
 # Initialize TANGO
+tango = Tango(device="cpu")
+tango.to("cuda")
 @spaces.GPU(duration=240)
 def gradio_generate(prompt, steps, guidance):