graphcast icon indicating copy to clipboard operation
graphcast copied to clipboard

Sharing models on the Hugging Face Hub

Open osanseviero opened this issue 1 year ago • 1 comments

Hi there!

At the moment, models are downloaded through a public GCP Bucket. It would be useful to have the models on the Hugging Face Hub (maybe under the https://hf.co/google organization?). It has some benefits as:

  • More visibility + discoverability of the existing models
  • Programmatic model access (either via wget or huggingface_hub if you want some of our caching mechanisms)
  • Download stats

Here is a guide in case you're interested, and I'm happy to guide you in the process.

Omar :hugs:

osanseviero avatar Nov 21 '23 13:11 osanseviero

Until they're ready to upload it to Huggingface Hub themselves, here's an unofficial port to Huggingface Hub:

Huggingface Collection: https://huggingface.co/collections/shermansiu/deepmind-graphcast-658be928925aadd433065afd Example Jupyter Notebook: https://github.com/shermansiu/graphcast/blob/main/graphcast_demo_hf.ipynb

Model weights, example inputs, and normalization statistics are all included.

I've tested the code, but please let me know if anything needs to be fixed.

shermansiu avatar Dec 29 '23 02:12 shermansiu