Update README.md

README.md

@@ -6,7 +6,7 @@ license: cc-by-nc-sa-4.0
 Original repo: https://github.com/google-deepmind/graphcast
 Original files are from this Google Cloud Bucket: https://console.cloud.google.com/storage/browser/dm_graphcast
 
-This repo contains the weights for `GraphCast`, the high-resolution model used in the GraphCast paper (0.25 degree resolution, 37 pressure levels), trained on ERA5 data from 1979 to 2017.
+This repo contains the weights for `GraphCast_small`, a smaller, low-resolution version of GraphCast (1 degree resolution, 13 pressure levels, and a smaller mesh), trained on ERA5 data from 1979 to 2015, useful to run a model with lower memory and compute constraints.
 
 ## License and Attribution
 The model weights are released by Google DeepMind.
@@ -20,8 +20,8 @@ You can load the model like so:
 ```python
 from graphcast import checkpoint
 
-REPO_ID = "shermansiu/dm_graphcast"
-FILENAME = ""
+REPO_ID = "shermansiu/dm_graphcast_small"
+FILENAME = "GraphCast_small - ERA5 1979-2015 - resolution 1.0 - pressure levels 13 - mesh 2to5 - precipitation input and output.npz"
 
 with open(hf_hub_download(repo_id=REPO_ID, filename=FILENAME), "rb") as f:
     ckpt = checkpoint.load(f, graphcast.CheckPoint)