Uploading Models
Last updated
Last updated
To upload models to the Hub, youโll need to create an account at BOINC AI. Models on the Hub are Git-based repositories, which give you versioning, branches, discoverability and sharing features, integration with over a dozen libraries, and more! You have control over what you want to upload to your repository, which could include checkpoints, configs, and any other files.
You can link repositories with an individual, such as osanseviero/fashion_brands_patterns, or with an organization, such as facebook/bart-large-xsum. Organizations can collect models related to a company, community, or library! If you choose an organization, the model will be featured on the organizationโs page, and every member of the organization will have the ability to contribute to the repository. You can create a new organization here.
There are several ways to upload models to the Hub, described below. We suggest adding a Model Card to your repo to document your model.
To create a brand new model repository, visit boincai/new. Then follow these steps:
In the โFiles and versionsโ tab, select โAdd Fileโ and specify โUpload Fileโ:
From there, select a file from your computer to upload and leave a helpful commit message to know what you are uploading:
Afterwards, click Commit changes to upload your model to the Hub!
Inspect files and history
You can check your repository with all the recently added files!
The UI allows you to explore the model files and commits and to see the diff introduced by each commit:
Add metadata
You can add metadata to your model card. You can specify:
the type of task this model is for, enabling widgets and the Inference API.
the used library (transformers
, spaCy
, etc.)
the language
the dataset
metrics
license
a lot more!
Read more about model tags here.
Add TensorBoard traces
Any repository that contains TensorBoard traces (filenames that contain tfevents
) is categorized with the TensorBoard
tag. As a convention, we suggest that you save traces under the runs/
subfolder. The โTraining metricsโ tab then makes it easy to review charts of the logged variables, like the loss or the accuracy.
Models trained with ๐ Transformers will generate TensorBoard traces by default if tensorboard
is installed.
Since model repos are just Git repositories, you can use Git to push your model files to the Hub. Follow the guide on Getting Started with Repositories to learn about using the git
CLI to commit and push your models.
The rich feature set in the boincai_hub
library allows you to manage repositories, including creating repos and uploading models to the Model Hub. Visit the client libraryโs documentation to learn more.