search

huggingface / huggingface.js

Utilities to use the Hugging Face Hub API
https://hf.co/docs/huggingface.js
MIT License
1.29k stars 160 forks source link

✨ Count downloads for audiocraft #773

Closed nateraw closed 1 week ago

nateraw commented 1 week ago

Hey there, wanted to add download counts for audiocraft to see how many folks are downloading songstarter-v0.2. :)

My model doesn't have transformers ckpts in it, so the counts are missed. Existing musicgen models from FB are only being counted if the user is using transformers too I think.

So I updated the logic here to count on state_dict.bin, which is audiocraft checkpoint, or config.json, which captures downloads from transformers.

julien-c commented 1 week ago

cool! and i gave a like to songstarter-v0.2 😁