From d7aa50917f1e2a2f79f2f00673afaaffe131b387 Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Wed, 25 Dec 2024 12:07:50 +0530 Subject: [PATCH] fixes --- finetrainers/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/finetrainers/trainer.py b/finetrainers/trainer.py index 04f5737..09afc1b 100644 --- a/finetrainers/trainer.py +++ b/finetrainers/trainer.py @@ -954,7 +954,7 @@ class Trainer: if self.args.push_to_hub: repo_id = self.args.hub_model_id or Path(self.args.output_dir).name - self.state.repo_id = create_repo(token=self.args.hub_token, name=repo_id).repo_id + self.state.repo_id = create_repo(token=self.args.hub_token, repo_id=repo_id).repo_id def _move_components_to_device(self): if self.text_encoder is not None: