Update app.py

app.py

@@ -290,7 +290,7 @@ def generate_music(
         for npy in stage1_output_set:
             codec_result = np.load(npy)
             decodec_rlt = []
-            with torch.inference_mode(), torch.autocast(device_type='cuda', dtype=torch.float16):
+            with torch.grad():
                 decoded_waveform = codec_model.decode(
                     torch.as_tensor(codec_result.astype(np.int16), dtype=torch.long).unsqueeze(0).permute(1, 0, 2).to(
                         device))
@@ -421,23 +421,22 @@ with gr.Blocks() as demo:
                 [
                     "Bass Metalcore Thrash Metal Furious bright vocal male Angry aggressive vocal Guitar",
                     """[verse]
-    Step back cause I'll ignite
-    Won't quit without a fight
-    No escape, gear up, it's a fierce fight
-    Brace up, raise your hands up and light
-    Fear the might. Step back cause I'll ignite
-    Won't back down without a fight
-    It keeps going and going, the heat is on.
-    
-    [chorus]
-    Hot flame. Hot flame.
-    Still here, still holding aim
-    I don't care if I'm bright or dim: nah.
-    I've made it clear, I'll make it again
-    All I want is my crew and my gain.
-    I'm feeling wild, got a bit of rebel style.
-    Locked inside my mind, hot flame.
+Step back cause I'll ignite
+Won't quit without a fight
+No escape, gear up, it's a fierce fight
+Brace up, raise your hands up and light
+Fear the might. Step back cause I'll ignite
+Won't back down without a fight
+It keeps going and going, the heat is on.
 
+[chorus]
+Hot flame. Hot flame.
+Still here, still holding aim
+I don't care if I'm bright or dim: nah.
+I've made it clear, I'll make it again
+All I want is my crew and my gain.
+I'm feeling wild, got a bit of rebel style.
+Locked inside my mind, hot flame.
                     """
                 ],
                 [