Skip to content

Commit 15d7071

Browse files
authored
Update terratorch library (#1731)
Adding `.ckpt` file extension because some supported models on the Hub do not use `.pt`, e.g. https://huggingface.co/ibm-granite/granite-geospatial-biomass/tree/main
1 parent 7d4e8d4 commit 15d7071

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

packages/tasks/src/model-libraries.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1072,7 +1072,7 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
10721072
repoUrl: "https://github.com/IBM/terratorch",
10731073
docsUrl: "https://ibm.github.io/terratorch/",
10741074
filter: false,
1075-
countDownloads: `path_extension:"pt"`,
1075+
countDownloads: `path_extension:"pt" OR path_extension:"ckpt"`,
10761076
snippets: snippets.terratorch,
10771077
},
10781078
"tic-clip": {

0 commit comments

Comments
 (0)