"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "a8817371c9c9f72882b37c4c646877b2032dcbfc"
Unverified Commit da7aabf2 authored by Philipp Schmid's avatar Philipp Schmid Committed by GitHub
Browse files

add str hub token to repository when provided else fallback to default (#14682)

* add str hub token to repository when provided else fallback to default True

* make style
parent 7375758b
...@@ -68,8 +68,11 @@ class PushToHubCallback(Callback): ...@@ -68,8 +68,11 @@ class PushToHubCallback(Callback):
hub_model_id = output_dir.absolute().name hub_model_id = output_dir.absolute().name
if "/" not in hub_model_id: if "/" not in hub_model_id:
hub_model_id = get_full_repo_name(hub_model_id, token=hub_token) hub_model_id = get_full_repo_name(hub_model_id, token=hub_token)
self.output_dir = output_dir self.output_dir = output_dir
self.repo = Repository(str(output_dir), clone_from=hub_model_id) self.repo = Repository(
str(output_dir), clone_from=hub_model_id, use_auth_token=hub_token if hub_token else True
)
self.tokenizer = tokenizer self.tokenizer = tokenizer
self.last_job = None self.last_job = None
self.checkpoint = checkpoint self.checkpoint = checkpoint
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment