diff --git a/examples/pytorch/diffusion_model/diffusers/framepack/main.py b/examples/pytorch/diffusion_model/diffusers/framepack/main.py index 0c25b079760..18fc7421fe7 100644 --- a/examples/pytorch/diffusion_model/diffusers/framepack/main.py +++ b/examples/pytorch/diffusion_model/diffusers/framepack/main.py @@ -358,8 +358,7 @@ def callback(d): text_encoder_2.requires_grad_(False) image_encoder.requires_grad_(False) - if not os.path.exists(args.output_video_path): - os.makedirs(args.output_video_path) + os.makedirs(args.output_video_path, exist_ok=True) init_process_group(backend="nccl") local_rank = int(os.environ["LOCAL_RANK"])