Update app.py

app.py

@@ -124,7 +124,7 @@ def generate_music(
         genre_txt=None,
         lyrics_txt=None,
         run_n_segments=3,
-        max_new_tokens=50,
+        max_new_tokens=45,
         use_audio_prompt=False,
         audio_prompt_path="",
         prompt_start_time=0.0,
@@ -162,7 +162,7 @@ def generate_music(
         # Decoding config
         top_p = 0.93
         temperature = 1.0
-        repetition_penalty = 1.2
+        repetition_penalty = 1.1
         start_of_segment = mmtokenizer.tokenize('[start_of_segment]')
         end_of_segment = mmtokenizer.tokenize('[end_of_segment]')
 
@@ -221,7 +221,7 @@ def generate_music(
                         logits_processor=LogitsProcessorList([BlockTokenRangeProcessor(0, 32002), BlockTokenRangeProcessor(32016, 32016)]),
                         guidance_scale=guidance_scale,
                         use_cache=True,
-                        num_beams=1
+                        num_beams = 3 if i > 1 else 1
                     )
                     if output_seq[0][-1].item() != mmtokenizer.eoa:
                         tensor_eoa = torch.as_tensor([[mmtokenizer.eoa]]).to(model.device)
@@ -368,10 +368,10 @@ Woke up in the morning, sun is shining bright
 Chasing all my dreams, gotta get my mind right
 City lights are fading, but my vision's clear
 Got my team beside me, no room for fear
-Walking through the streets, beats inside my head
-Every step I take, closer to the bread
 
 [chorus]
+Walking through the streets, beats inside my head
+Every step I take, closer to the bread
 This is my life, and I'm aiming for the top
 Never gonna quit, no, I'm never gonna stop
 Through the highs and lows, I'mma keep it real