Update app.py

app.py

@@ -87,8 +87,6 @@ model = AutoModelForCausalLM.from_pretrained(
 # assistant_model.eval()
 model.eval()
 
-early_exit_layer = 4
-
 basic_model_config = './xcodec_mini_infer/final_ckpt/config.yaml'
 resume_path = './xcodec_mini_infer/final_ckpt/ckpt_00360000.pth'
 config_path = './xcodec_mini_infer/decoders/config.yaml'
@@ -234,13 +232,11 @@ def generate_music(
                     repetition_penalty=repetition_penalty,
                     eos_token_id=mmtokenizer.eoa,
                     pad_token_id=mmtokenizer.eoa,
-                    logits_processor=LogitsProcessorList([BlockTokenRangeProcessor(0, 32002),
-                                                        BlockTokenRangeProcessor(32016, 32016)]),
+                    # logits_processor=LogitsProcessorList([BlockTokenRangeProcessor(0, 32002), BlockTokenRangeProcessor(32016, 32016)]),
                     guidance_scale=guidance_scale,
                     use_cache=True,
                     top_k=50,
-                    num_beams=1,
-                    assistant_early_exit=early_exit_layer
+                    num_beams=1
                 )
                 if output_seq[0][-1].item() != mmtokenizer.eoa:
                     tensor_eoa = torch.as_tensor([[mmtokenizer.eoa]]).to(model.device)