Feat(config): Add hub_strategy (#386)

This commit is contained in:
NanoCode012
2023-08-14 20:12:55 +09:00
committed by GitHub
parent 63fdb5a7fb
commit 73a0b6ead5
2 changed files with 6 additions and 0 deletions

View File

@@ -364,6 +364,9 @@ dataset_prepared_path: data/last_run_prepared
push_dataset_to_hub: # repo path
# push checkpoints to hub
hub_model_id: # repo path to push finetuned model
# how to push checkpoints to hub
# https://huggingface.co/docs/transformers/v4.31.0/en/main_classes/trainer#transformers.TrainingArguments.hub_strategy
hub_strategy:
# whether to use hf `use_auth_token` for loading datasets. Useful for fetching private datasets
# required to be true when used in combination with `push_dataset_to_hub`
hf_use_auth_token: # boolean