google-deepmind / graphcast

Apache License 2.0
4.37k stars 538 forks source link

Sharing models on the Hugging Face Hub #23

Open osanseviero opened 7 months ago

osanseviero commented 7 months ago

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:

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

Omar :hugs:

shermansiu commented 6 months ago

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.