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