Huggingface models: how to store a different version of a model

Viewed 15

I have a model that I pushed to the remote using the following code:

from transformers import CLIPProcessor, CLIPModel

checkpoint = "./checkpoints-15/checkpoint-60"
    
model = CLIPModel.from_pretrained(checkpoint)
processor = CLIPProcessor.from_pretrained(checkpoint)

repo = "vincentclaes/emoji-predictor"
model.push_to_hub(repo, use_temp_dir=True)
processor.push_to_hub(repo, use_temp_dir=True)

On the UI I see my model under a main branch: enter image description here

What if I want to store multiple versions of a model?

  • Can I create a separate git branch?
  • Can I create a git tag?

How do I do this using the huggingface tools? Thinking transformers, huggingface_hub, ...

0 Answers
Related