Unverified Commit 159885ad authored by Sayak Paul's avatar Sayak Paul Committed by GitHub
Browse files

correct hub_token exposition behaviour (thanks to @bghira). (#6918)

parent 7337eea5
...@@ -584,6 +584,12 @@ class TextualInversionDataset(Dataset): ...@@ -584,6 +584,12 @@ class TextualInversionDataset(Dataset):
def main(): def main():
args = parse_args() args = parse_args()
if args.report_to == "wandb" and args.hub_token is not None:
raise ValueError(
"You cannot use both --report_to=wandb and --hub_token due to a security risk of exposing your token."
" Please use `huggingface-cli login` to authenticate with the Hub."
)
logging_dir = os.path.join(args.output_dir, args.logging_dir) logging_dir = os.path.join(args.output_dir, args.logging_dir)
accelerator_project_config = ProjectConfiguration(project_dir=args.output_dir, logging_dir=logging_dir) accelerator_project_config = ProjectConfiguration(project_dir=args.output_dir, logging_dir=logging_dir)
accelerator = Accelerator( accelerator = Accelerator(
......
...@@ -442,6 +442,12 @@ def parse_args(): ...@@ -442,6 +442,12 @@ def parse_args():
def main(): def main():
args = parse_args() args = parse_args()
if args.report_to == "wandb" and args.hub_token is not None:
raise ValueError(
"You cannot use both --report_to=wandb and --hub_token due to a security risk of exposing your token."
" Please use `huggingface-cli login` to authenticate with the Hub."
)
logging_dir = os.path.join(args.output_dir, args.logging_dir) logging_dir = os.path.join(args.output_dir, args.logging_dir)
accelerator_project_config = ProjectConfiguration( accelerator_project_config = ProjectConfiguration(
total_limit=args.checkpoints_total_limit, project_dir=args.output_dir, logging_dir=logging_dir total_limit=args.checkpoints_total_limit, project_dir=args.output_dir, logging_dir=logging_dir
......
...@@ -440,6 +440,12 @@ def parse_args(): ...@@ -440,6 +440,12 @@ def parse_args():
def main(): def main():
args = parse_args() args = parse_args()
if args.report_to == "wandb" and args.hub_token is not None:
raise ValueError(
"You cannot use both --report_to=wandb and --hub_token due to a security risk of exposing your token."
" Please use `huggingface-cli login` to authenticate with the Hub."
)
logging_dir = os.path.join(args.output_dir, args.logging_dir) logging_dir = os.path.join(args.output_dir, args.logging_dir)
accelerator_project_config = ProjectConfiguration( accelerator_project_config = ProjectConfiguration(
total_limit=args.checkpoints_total_limit, project_dir=args.output_dir, logging_dir=logging_dir total_limit=args.checkpoints_total_limit, project_dir=args.output_dir, logging_dir=logging_dir
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment