Update inference/infer.py

inference/infer.py

@@ -141,6 +141,9 @@ 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(enumerate(tqdm(prompt_texts[: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