Update accel_sdxl_gen_img.py

This commit is contained in:
DKnight54
2025-02-08 10:30:31 +08:00
committed by GitHub
parent 8d14893598
commit 984441c014

View File

@@ -2389,7 +2389,10 @@ def main(args):
all_seeds = gather_object(seeds)
all_clip_prompts = gather_object(clip_prompts)
all_raw_prompts = gather_object(raw_prompts)
all_init_images = gather_object(init_images)
if init_images not None:
all_init_images = gather_object(init_images)
else:
all_init_images = None
if distributed_state.is_main_process:
highres_prefix = ("0" if highres_1st else "1") if highres_fix else ""