Update accel_sdxl_gen_img.py

This commit is contained in:
DKnight54
2025-02-01 21:50:58 +08:00
committed by GitHub
parent 03232a6194
commit ff77d3afe1

View File

@@ -2907,7 +2907,7 @@ def main(args):
temp_list = []
for ext_batch in batch_separated_list:
for i in range(distributed_state.num_processes):
templist.append(ext_batch[i :: distributed_state.num_processes])
temp_list.append(ext_batch[i :: distributed_state.num_processes])
logger.info(f"templist: {len(temp_list)}")
batch_separated_list = []
for sub_batch_list in temp_list: