Update app.py

app.py

@@ -158,8 +158,8 @@ def generate_music(
         raw_output = None
 
         # Decoding config
-        top_p = 0.9
-        temperature = 0.9
+        top_p = 0.93
+        temperature = 1.0
         repetition_penalty = 1.2
         start_of_segment = mmtokenizer.tokenize('[start_of_segment]')
         end_of_segment = mmtokenizer.tokenize('[end_of_segment]')
@@ -220,7 +220,7 @@ def generate_music(
                     logits_processor=LogitsProcessorList([BlockTokenRangeProcessor(0, 32002), BlockTokenRangeProcessor(32016, 32016)]),
                     guidance_scale=guidance_scale,
                     use_cache=True,
-                    num_beams=3
+                    num_beams=1
                 )
                 if output_seq[0][-1].item() != mmtokenizer.eoa:
                     tensor_eoa = torch.as_tensor([[mmtokenizer.eoa]]).to(model.device)