KingNish commited on
Commit
28e5ba2
·
verified ·
1 Parent(s): c3d0f2b

Update inference/infer.py

Browse files
Files changed (1) hide show
  1. inference/infer.py +1 -1
inference/infer.py CHANGED
@@ -142,7 +142,7 @@ end_of_segment = mmtokenizer.tokenize('[end_of_segment]')
142
  # Format text prompt
143
  run_n_segments = min(args.run_n_segments+1, len(lyrics))
144
 
145
- print(enumerate(tqdm(prompt_texts[:run_n_segments])))
146
 
147
  for i, p in enumerate(tqdm(prompt_texts[:run_n_segments])):
148
  section_text = p.replace('[start_of_segment]', '').replace('[end_of_segment]', '')
 
142
  # Format text prompt
143
  run_n_segments = min(args.run_n_segments+1, len(lyrics))
144
 
145
+ print(list(enumerate(tqdm(prompt_texts[:run_n_segments]))))
146
 
147
  for i, p in enumerate(tqdm(prompt_texts[:run_n_segments])):
148
  section_text = p.replace('[start_of_segment]', '').replace('[end_of_segment]', '')