Spaces:

declare-lab
/

mustango

Running on A10G

deepanway commited on Nov 17, 2023

Commit

b3bab00

•

1 Parent(s): 9e0eee2

update files for device agnostic inference

Files changed (4) hide show

.ipynb_checkpoints/app-checkpoint.py CHANGED Viewed

@@ -216,11 +216,11 @@ class Mustango:
 # Initialize Mustango
 if torch.cuda.is_available():
-    mustango = Mustango(device="cpu")
 else:
     mustango = Mustango(device="cpu")
-output_wave = mustango.generate("This techno song features a synth lead playing the main melody.", 5, 3, disable_progress=False)
 def gradio_generate(prompt, steps, guidance):
     output_wave = mustango.generate(prompt, steps, guidance)

 # Initialize Mustango
 if torch.cuda.is_available():
+    mustango = Mustango()
 else:
     mustango = Mustango(device="cpu")
+# output_wave = mustango.generate("This techno song features a synth lead playing the main melody.", 5, 3, disable_progress=False)
 def gradio_generate(prompt, steps, guidance):
     output_wave = mustango.generate(prompt, steps, guidance)

__pycache__/models.cpython-310.pyc CHANGED Viewed

Binary files a/__pycache__/models.cpython-310.pyc and b/__pycache__/models.cpython-310.pyc differ

app.py CHANGED Viewed

@@ -216,11 +216,11 @@ class Mustango:
 # Initialize Mustango
 if torch.cuda.is_available():
-    mustango = Mustango(device="cpu")
 else:
     mustango = Mustango(device="cpu")
-output_wave = mustango.generate("This techno song features a synth lead playing the main melody.", 5, 3, disable_progress=False)
 def gradio_generate(prompt, steps, guidance):
     output_wave = mustango.generate(prompt, steps, guidance)

 # Initialize Mustango
 if torch.cuda.is_available():
+    mustango = Mustango()
 else:
     mustango = Mustango(device="cpu")
+# output_wave = mustango.generate("This techno song features a synth lead playing the main melody.", 5, 3, disable_progress=False)
 def gradio_generate(prompt, steps, guidance):
     output_wave = mustango.generate(prompt, steps, guidance)

layers/__pycache__/layers.cpython-310.pyc CHANGED Viewed

Binary files a/layers/__pycache__/layers.cpython-310.pyc and b/layers/__pycache__/layers.cpython-310.pyc differ