Add library metadata to HF integration

naver / dust3r

DUSt3R: Geometric 3D Vision Made Easy

https://dust3r.europe.naverlabs.com/

Other

4.65k stars 515 forks source link

Add library metadata to HF integration #79

Closed Wauplin closed 2 months ago

Wauplin commented 2 months ago

Follow-up PR after https://github.com/naver/dust3r/pull/78.

The HF integration allows to define high level metadata valid for all models pushed to the Hub. This allows a better discoverability as users can find the model based on tags (either library name or task). This PR aims at adding these metadata while ensuring the HF integration is still optional.

Please let me know if you have any questions!

yocabon commented 2 months ago

Hi, thanks. I just realized that huggingface_hub is already required by gradio... So there's very little point in making it optional if it has to be installed anyway. I'll change that.

Wauplin commented 2 months ago

Ah yeah, good catch! FYI huggingface_hub aims at staying as lightweight as possible when it comes to dependencies (current deps are tqdm, requests, filelock and fsspec). But yeah if gradio is installed then you'll have it automatically

Thanks for merging the PR btw!