facebookresearch / facebookresearch/EdgeTAM

Improve HF integration

Open
#4 1 comment 0 reactions 0 assignees View on GitHub
Dominant language
Jupyter Notebook
Stars
978
Forks
83
PR merge metrics
No merged PRs in 30d

Description

Hi,

Niels here from the open-source team at Hugging Face. Thanks for making the model and demo available on the hub :)

We could also integrate loading checkpoints from the hub using a `from_pretrained` method as we did for SAM-2: https://github.com/facebookresearch/sam2?tab=readme-ov-file#load-from--hugging-face. The checkpoints are downloaded quite a lot: https://huggingface.co/models?sort=downloads&search=facebook%2Fsam2.

I see currently downloads aren't working for your model: https://huggingface.co/facebook/EdgeTAM, so integrating with the hub would solve that.

Let me know if you are interested.

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.