graphcast
graphcast copied to clipboard
Sharing models on the Hugging Face Hub
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
wgetorhuggingface_hubif 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:
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.