Hugging Face's logo

ibm-granite
/
granite-geospatial-canopyheight

PyTorch
Model card Files Community
4
romeokienzler commited on
Commit
ced37d8
•
1 Parent(s): ceb7022

fix checkpoint file name

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -28,7 +28,7 @@ You can use it easily with Terratorch through:
28
  ```python
29
  from terratorch.cli_tools import LightningInferenceModel
30
 
31
- ckpt_path = hf_hub_download(repo_id="ibm-granite/granite-geospatial-canopyheight", filename="canopyheight_model.ckpt")
32
  config_path = hf_hub_download(repo_id="ibm-granite/granite-geospatial-canopyheight", filename="config.yaml")
33
 
34
  model = LightningInferenceModel.from_config(config_path, ckpt_path)
 
28
  ```python
29
  from terratorch.cli_tools import LightningInferenceModel
30
 
31
+ ckpt_path = hf_hub_download(repo_id="ibm-granite/granite-geospatial-canopyheight", filename="pytorch_model.bin")
32
  config_path = hf_hub_download(repo_id="ibm-granite/granite-geospatial-canopyheight", filename="config.yaml")
33
 
34
  model = LightningInferenceModel.from_config(config_path, ckpt_path)