You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Is your feature request related to a problem? Please describe.
When training with FARMReader finishes, there should be an option to upload it automatically to HF hub.
Describe the solution you'd like
Add option and template to auto upload trained / fine-tuned model to HF.
Describe alternatives you've considered
Implement helper function to upload model to HF manually. Maybe we go for that if the selection of the model snapshot involves human interaction.
Tasks:
add hf_token, hf_repo, hf_branch, hf_model_card params to FARMReader.train(), DensePassageRetriever.train() and TableTextRetriever.train()
create repo if not exists
convert model into huggingface format
upload model files and model card
provide model card template
write documentation about how to use it
The text was updated successfully, but these errors were encountered:
Using huggingface_hub lib we can create a repo and upload them model files. To add a model card, we simply have to upload a README.md according to this structure.
Additionally we might want to enable users to specify a version of the model. This could be achieved by using branches in the model repo.
I got a first draft of this to work with FarmReader. The model gets converted to the transformers model format so it can be reused later by loading it back from HF too. Created a draft PR 😊
Is your feature request related to a problem? Please describe.
When training with FARMReader finishes, there should be an option to upload it automatically to HF hub.
Describe the solution you'd like
Add option and template to auto upload trained / fine-tuned model to HF.
Describe alternatives you've considered
Implement helper function to upload model to HF manually. Maybe we go for that if the selection of the model snapshot involves human interaction.
Tasks:
hf_token
,hf_repo
,hf_branch
,hf_model_card
params toFARMReader.train()
,DensePassageRetriever.train()
andTableTextRetriever.train()
The text was updated successfully, but these errors were encountered: