Add simple debug check for raw_output variable

- Add debug information when raw_output is not defined
- Print generation loop parameters for troubleshooting
- Provide clearer error message for generation failures

YuEGP/inference/infer.py

@@ -316,6 +316,14 @@ for i, p in enumerate(tqdm(prompt_texts[:run_n_segments])):
             raise RuntimeError(f"Generation failed at segment {i}: {e}")
 
 # save raw output and check sanity
+# Check if raw_output was defined (debug for generation issues)
+if 'raw_output' not in locals():
+    print("❌ ERROR: raw_output was not defined. This indicates the generation loop failed or didn't execute properly.")
+    print(f"run_n_segments: {run_n_segments}")
+    print(f"len(lyrics): {len(lyrics)}")
+    print(f"prompt_texts length: {len(prompt_texts)}")
+    raise RuntimeError("Generation failed - raw_output not defined")
+    
 ids = raw_output[0].cpu().numpy()
 soa_idx = np.where(ids == mmtokenizer.soa)[0].tolist()
 eoa_idx = np.where(ids == mmtokenizer.eoa)[0].tolist()