Spaces:

TheStageAI
/

Elastic-musicgen-large

Running on L40S

quazim commited on Jun 27

Commit

9c28790

1 Parent(s): fa9bb88

updated

Files changed (1) hide show

app.py CHANGED Viewed

@@ -90,7 +90,7 @@ def generate_music(text_prompt, duration=10, temperature=1.0, top_k=250, top_p=0
             return_tensors="pt",
         ).to("cuda")
         print("Processor end")
-        print(inputs.device)
         # Generate audio
         with torch.no_grad():

             return_tensors="pt",
         ).to("cuda")
         print("Processor end")
+        print(inputs)
         # Generate audio
         with torch.no_grad():