Skip to content

Commit 27b995d

Browse files
julian-qWauplin
andauthored
Track oasis .safetensors instead of .pt (#1018)
Hello! We recently updated our model (https://huggingface.co/Etched/oasis-500m/tree/main) to use the safetensors format instead of pickle. We want to make the .safetensors the default and remove the .pt, and so this PR allows download counting on the .safetensors instead. Co-authored-by: Lucain <[email protected]>
1 parent 4606df3 commit 27b995d

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
@@ -480,7 +480,7 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
480480
prettyLabel: "open-oasis",
481481
repoName: "open-oasis",
482482
repoUrl: "https://github.com/etched-ai/open-oasis",
483-
countDownloads: `path:"oasis500m.pt"`,
483+
countDownloads: `path:"oasis500m.safetensors"`,
484484
},
485485
open_clip: {
486486
prettyLabel: "OpenCLIP",

0 commit comments

Comments
 (0)