Feat: huggingface hub push

This commit is contained in:
litagin02
2024-02-17 11:01:16 +09:00
parent 19f352f9c8
commit ff730476e7
2 changed files with 34 additions and 3 deletions

View File

@@ -84,6 +84,11 @@ def run():
action="store_true",
help="Speed up training by disabling logging and evaluation.",
)
parser.add_argument(
"--repo_id",
help="Huggingface model repo id to backup the model.",
default=None,
)
args = parser.parse_args()
# Set log file