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

Add model to HuggingFace #4

Open
NielsRogge opened this issue Oct 11, 2021 · 3 comments
Open

Add model to HuggingFace #4

NielsRogge opened this issue Oct 11, 2021 · 3 comments

Comments

@NielsRogge
Copy link

Hi,

Really cool work! One thing that could be cool (it's a feature that was recently added to HuggingFace Transformers) is the ability to load any custom PyTorch model from the hub as introduced in this PR. The idea would be that you can create a new model on the hub (it's as easy as clicking "new model", then git add ., git commit, git push to upload the weights to the repo on the hub). You can also pass in a custom script defining your model. Next, one can load your custom model as follows:

from transformers import AutoModel

model = AutoModel.from_pretrained("rmokday/clip_prefix_caption", trust_remote_code=True)

This could be easier compared to hosting the models on Google Drive, and you don't need to define the model again everytime someone wants to use it in a notebook :)

Do you mind trying this out?

Kind regards,

Niels

@rmokady
Copy link
Owner

rmokady commented Oct 11, 2021

@NielsRogge Sounds cool, I'll hope to try it out soon

@RylanSchaeffer
Copy link

@rmokady any updates on this?

@rmokady
Copy link
Owner

rmokady commented Oct 21, 2021

@rmokady any updates on this?

Currently our top priority is to improve our model, so I guess it will be delayed
If someone will find it relevant, we will very appreciate this contribution

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

No branches or pull requests

3 participants