Update inference/infer.py

inference/infer.py

@@ -144,6 +144,7 @@ start_of_segment = mmtokenizer.tokenize('[start_of_segment]')
 end_of_segment = mmtokenizer.tokenize('[end_of_segment]')
 # Format text prompt
 run_n_segments = min(args.run_n_segments+1, len(lyrics))
+print(f"RUN N SEGMENTS: {run_n_segments}")
 for i, p in enumerate(tqdm(prompt_texts[:run_n_segments])):
     section_text = p.replace('[start_of_segment]', '').replace('[end_of_segment]', '')
     guidance_scale = 1.5 if i <=1 else 1.2