Update inference/infer.py
Browse files- 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]', '')
|