Initializes a git repo in `self.args.hub_model_id`.
(self, token: str | None = None)
| 3928 | # ---- Hub Integration ---- |
| 3929 | |
| 3930 | def init_hf_repo(self, token: str | None = None) -> None: |
| 3931 | """ |
| 3932 | Initializes a git repo in `self.args.hub_model_id`. |
| 3933 | """ |
| 3934 | # Only on process zero |
| 3935 | if not self.is_world_process_zero(): |
| 3936 | return |
| 3937 | |
| 3938 | if self.args.hub_model_id is None: |
| 3939 | repo_name = Path(self.args.output_dir).absolute().name |
| 3940 | else: |
| 3941 | repo_name = self.args.hub_model_id |
| 3942 | |
| 3943 | token = token if token is not None else self.args.hub_token |
| 3944 | repo_url = hf_api().create_repo(repo_name, token=token, private=self.args.hub_private_repo, exist_ok=True) |
| 3945 | self.hub_model_id = repo_url.repo_id |
| 3946 | self.push_in_progress = None |
| 3947 | |
| 3948 | def create_model_card( |
| 3949 | self, |
no test coverage detected