storytold/FineTrainers-Conditioning
2
0
Fork 0
mirror of https://github.com/storytold/FineTrainers-Conditioning.git synced 2026-10-17 00:23:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

fixes

Browse Source
This commit is contained in:
sayakpaul
2024-12-25 12:07:50 +05:30
parent 69b585a992
commit d7aa50917f
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
finetrainers/trainer.py
+1 -1
View File
@@ -954,7 +954,7 @@ class Trainer:
if self.args.push_to_hub: if self.args.push_to_hub:
repo_id = self.args.hub_model_id or Path(self.args.output_dir).name 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): def _move_components_to_device(self):
if self.text_encoder is not None: if self.text_encoder is not None: