chore: add perf mode to example (#319)

* chore: add perf mode to example

* chore: add perf mode to example

Author: DefTruth

examples/parallelism/run_qwen_image_cp.py

@@ -89,16 +89,17 @@
 
 def run_pipe():
     # do_true_cfg = true_cfg_scale > 1 and has_neg_prompt
-    image = pipe(
+    output = pipe(
         prompt=prompt + positive_magic["en"],
         negative_prompt=negative_prompt,
         width=1024 if args.width is None else args.width,
         height=1024 if args.height is None else args.height,
         num_inference_steps=50 if args.steps is None else args.steps,
         true_cfg_scale=4.0,
         generator=torch.Generator(device="cpu").manual_seed(42),
-    ).images[0]
-
+        output_type="latent" if args.perf else "pil",
+    )
+    image = output.images[0] if not args.perf else None
     return image
 
 
@@ -119,7 +120,8 @@ def run_pipe():
     time_cost = end - start
     save_path = f"qwen-image.{strify(args, pipe)}.png"
     print(f"Time cost: {time_cost:.2f}s")
-    print(f"Saving image to {save_path}")
-    image.save(save_path)
+    if not args.perf:
+        print(f"Saving image to {save_path}")
+        image.save(save_path)
 
 maybe_destroy_distributed()

examples/utils.py

@@ -66,6 +66,7 @@ def get_args(
         default=None,
         choices=[None, "flash", "_native_cudnn"],
     )
+    parser.add_argument("--perf", action="store_true", default=False)
     return parser.parse_args() if parse else parser