# The S3 directory you want to save the merged weights
s3_checkpoint_dir = "<s3_checkpoint_dir>"