diff --git a/accel_sdxl_gen_img.py b/accel_sdxl_gen_img.py index c923e2e6..a7cadcb4 100644 --- a/accel_sdxl_gen_img.py +++ b/accel_sdxl_gen_img.py @@ -1846,8 +1846,8 @@ def main(args): pipe.set_control_nets(control_nets) logger.info(f"pipeline on {device} is ready.") distributed_state.wait_for_everyone() - pipes = gather_object([pipe]) - unets = gather_object([unet]) + #pipes = gather_object([pipe]) + #unets = gather_object([unet]) if args.diffusers_xformers: pipe.enable_xformers_memory_efficient_attention()