Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Push trained model to the Hugging Face Hub #9

Open
wants to merge 1 commit into
base: main
Choose a base branch
from

Conversation

pcuenca
Copy link
Contributor

@pcuenca pcuenca commented Jun 30, 2023

Notes:

  • The name of the repo is currently taken from the output directory.
  • Only the final model is pushed. If pushing all checkpoints is preferred, we could do it as part of run_save_model.

Downloading a checkpoint from the Hub will be addressed in a follow-up PR.

Notes:
- The name of the repo is currently taken from the output directory.
- Only the final model is pushed. If pushing all checkpoints is
preferred, we could do it as part of `run_save_model`.

Downloading a checkpoint from the Hub will be addressed in a follow-up
PR.
@borisdayma
Copy link
Owner

Cool! I'll run a test before merging.
Can use it to push the base model we'll train with sigmoid loss.
Will prep the data first.

@pcuenca
Copy link
Contributor Author

pcuenca commented Jun 30, 2023

Awesome, let me know so we can work on loading from the Hub too :)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants