[diffusion] feat: support tp for qwen-image-edit-2511 (#18619)

Co-authored-by: Mick <mickjagger19@icloud.com>
2026-02-13 13:04:29 +08:00
parent d29e331491
commit 5700b19cbf
3 changed files with 109 additions and 24 deletions
@@ -218,7 +218,7 @@ class GPUWorker:

            # Save output to file and return file path only if requested. Avoid the serialization
            # and deserialization overhead between scheduler_client and gpu_worker.
-            if req.save_output and req.return_file_paths_only:
+            if req.save_output and req.return_file_paths_only and self.rank == 0:
                output_paths = save_outputs(
                    output_batch.output,
                    req.data_type,