Skip to content

Commit

Permalink
Update accel_sdxl_gen_img.py
Browse files Browse the repository at this point in the history
  • Loading branch information
DKnight54 authored Feb 1, 2025
1 parent 03232a6 commit ff77d3a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion accel_sdxl_gen_img.py
Original file line number Diff line number Diff line change
Expand Up @@ -2907,7 +2907,7 @@ def scale_and_round(x):
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:
Expand Down

0 comments on commit ff77d3a

Please sign in to comment.