Fix syntax errors in infer.py

- Remove incomplete try-except blocks that caused syntax errors
- Fix indentation issues in generation loop
- Clean up orphaned except clauses
- Restore proper code structure while keeping debug information

YuEGP/inference/infer.py

@@ -236,11 +236,10 @@ end_of_segment = mmtokenizer.tokenize('[end_of_segment]')
 # Format text prompt
 run_n_segments = min(args.run_n_segments + 1, len(lyrics))
 for i, p in enumerate(tqdm(prompt_texts[:run_n_segments])):
-    try:
-        section_text = p.replace('[start_of_segment]', '').replace('[end_of_segment]', '')
-        guidance_scale = 1.5 if i <= 1 else 1.2
-        if i == 0:
-            continue
+    section_text = p.replace('[start_of_segment]', '').replace('[end_of_segment]', '')
+    guidance_scale = 1.5 if i <= 1 else 1.2
+    if i == 0:
+        continue
     if i == 1:
         if args.use_dual_tracks_prompt or args.use_audio_prompt:
             if args.use_dual_tracks_prompt:
@@ -299,21 +298,10 @@ for i, p in enumerate(tqdm(prompt_texts[:run_n_segments])):
         if output_seq[0][-1].item() != mmtokenizer.eoa:
             tensor_eoa = torch.as_tensor([[mmtokenizer.eoa]]).to(model.device)
             output_seq = torch.cat((output_seq, tensor_eoa), dim=1)
-        if i > 1:
-            raw_output = torch.cat([raw_output, prompt_ids, output_seq[:, input_ids.shape[-1]:]], dim=1)
-        else:
-            raw_output = output_seq
-    except Exception as e:
-        print(f"❌ Error during generation for segment {i}: {e}")
-        print(f"Section text: {section_text}")
-        import traceback
-        traceback.print_exc()
-        # If this is the first successful generation, we still have something
-        if i > 1 and 'raw_output' in locals():
-            print("⚠️ Using partial output from previous segments")
-            break
-        else:
-            raise RuntimeError(f"Generation failed at segment {i}: {e}")
+    if i > 1:
+        raw_output = torch.cat([raw_output, prompt_ids, output_seq[:, input_ids.shape[-1]:]], dim=1)
+    else:
+        raw_output = output_seq
 
 # save raw output and check sanity
 # Check if raw_output was defined (debug for generation issues)