From 4f3b93a474787c6b87ac11721d921c41b2b58eec Mon Sep 17 00:00:00 2001 From: "Wang, Mengni" Date: Mon, 8 Dec 2025 16:56:27 +0800 Subject: [PATCH] Update main.py --- examples/pytorch/diffusion_model/diffusers/flux/main.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/examples/pytorch/diffusion_model/diffusers/flux/main.py b/examples/pytorch/diffusion_model/diffusers/flux/main.py index c9e0b98d5c0..f2c415cb4ca 100644 --- a/examples/pytorch/diffusion_model/diffusers/flux/main.py +++ b/examples/pytorch/diffusion_model/diffusers/flux/main.py @@ -122,8 +122,7 @@ def tune(): if args.inference: pipe = AutoPipelineForText2Image.from_pretrained(args.model, torch_dtype=torch.bfloat16) - if not os.path.exists(args.output_image_path): - os.makedirs(args.output_image_path) + os.makedirs(args.output_image_path, exist_ok=True) if os.path.exists(args.output_dir) and os.path.exists(os.path.join(args.output_dir, "diffusion_pytorch_model.safetensors.index.json")): print(f"Loading quantized model from {args.output_dir}")