Update inference/infer.py
Browse files- inference/infer.py +1 -0
inference/infer.py
CHANGED
|
@@ -144,6 +144,7 @@ start_of_segment = mmtokenizer.tokenize('[start_of_segment]')
|
|
| 144 |
end_of_segment = mmtokenizer.tokenize('[end_of_segment]')
|
| 145 |
# Format text prompt
|
| 146 |
run_n_segments = min(args.run_n_segments+1, len(lyrics))
|
|
|
|
| 147 |
for i, p in enumerate(tqdm(prompt_texts[:run_n_segments])):
|
| 148 |
section_text = p.replace('[start_of_segment]', '').replace('[end_of_segment]', '')
|
| 149 |
guidance_scale = 1.5 if i <=1 else 1.2
|
|
|
|
| 144 |
end_of_segment = mmtokenizer.tokenize('[end_of_segment]')
|
| 145 |
# Format text prompt
|
| 146 |
run_n_segments = min(args.run_n_segments+1, len(lyrics))
|
| 147 |
+
print(f"RUN N SEGMENTS: {run_n_segments}")
|
| 148 |
for i, p in enumerate(tqdm(prompt_texts[:run_n_segments])):
|
| 149 |
section_text = p.replace('[start_of_segment]', '').replace('[end_of_segment]', '')
|
| 150 |
guidance_scale = 1.5 if i <=1 else 1.2
|