nicolabs/Open-Sora
1
0
Fork 0
mirror of https://github.com/hpcaitech/Open-Sora.git synced 2026-05-21 11:59:01 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

[inference] renamed duplicated variables (#153)

Browse source
This commit is contained in:
Frank Lee 2024-06-17 21:45:58 +08:00 committed by GitHub
parent 67fa660db5
commit 9a5ba168fb
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
scripts/inference.py
View file

@ -221,11 +221,11 @@ def main():
# recover the prompt list
batched_prompt_segment_list = []
start_idx = 0
segment_start_idx = 0
all_prompts = broadcast_obj_list[0]
for num_segment in prompt_segment_length:
batched_prompt_segment_list.append(all_prompts[start_idx : start_idx + num_segment])
start_idx += num_segment
batched_prompt_segment_list.append(all_prompts[segment_start_idx : segment_start_idx + num_segment])
segment_start_idx += num_segment
# 2. append score
for idx, prompt_segment_list in enumerate(batched_prompt_segment_list):