Skip to content

Commit d69619c

Browse files
authored
Update accel_sdxl_gen_img.py
1 parent c6e6643 commit d69619c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

accel_sdxl_gen_img.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2946,7 +2946,7 @@ def scale_and_round(x):
29462946
ext_separated_list_of_batches[x] = holder[:]
29472947
'''
29482948
distributed_state.wait_for_everyone()
2949-
'''
2949+
29502950
if distributed_state.is_main_process:
29512951
batchlogstr = "Running through ext_separated_list_of_batches Before Gather:\n"
29522952
for x in range(len(ext_separated_list_of_batches)):
@@ -2956,7 +2956,7 @@ def scale_and_round(x):
29562956
for z in range(len(ext_separated_list_of_batches[x][y])):
29572957
batchlogstr += f" Image {z} of {len(ext_separated_list_of_batches[x][y])} break: {ext_separated_list_of_batches[x][y][z].global_count}\n"
29582958
logger.info(batchlogstr)
2959-
'''
2959+
29602960
ext_separated_list_of_batches = gather_object(ext_separated_list_of_batches)
29612961
del extinfo
29622962
#logger.info(f"\nDevice {distributed_state.device}: {ext_separated_list_of_batches}")

0 commit comments

Comments
 (0)