Skip to content

Commit

Permalink
only upload the hf_metadata with beaker job (#305)
Browse files Browse the repository at this point in the history
  • Loading branch information
vwxyzjn authored Aug 28, 2024
1 parent d950880 commit ade34d0
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions open_instruct/finetune.py
Original file line number Diff line number Diff line change
Expand Up @@ -1017,7 +1017,7 @@ def main(args: FlatArguments):
location=args.hf_repo_id,
hf_repo_revision=args.hf_repo_revision,
)
if args.hf_metadata_dataset:
if args.hf_metadata_dataset and accelerator.is_main_process and is_beaker_job():
# dpo script only supports these two options right now for datasets
if args.dataset_mixer:
dataset_list = args.dataset_mixer.keys()
Expand All @@ -1027,7 +1027,6 @@ def main(args: FlatArguments):
dataset_list = [args.dataset_name]
else:
dataset_list = [args.train_file]
beaker_config = maybe_get_beaker_config()
# mainly just focussing here on what would be useful for the leaderboard.
# wandb will have even more useful information.
metadata_blob = {
Expand Down

0 comments on commit ade34d0

Please sign in to comment.