Skip to content

Commit dd25c60

Browse files
authored
Add 4 explicit weight filenames for open_clip counts. Fix #1122 (#1123)
Not sure about my multi-line string break...
1 parent 1a07402 commit dd25c60

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

packages/tasks/src/model-libraries.ts

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -544,7 +544,10 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
544544
repoUrl: "https://github.com/mlfoundations/open_clip",
545545
snippets: snippets.open_clip,
546546
filter: true,
547-
countDownloads: `path_extension:"bin" AND path_filename:*pytorch_model`,
547+
countDownloads: `path:"open_clip_model.safetensors"
548+
OR path:"model.safetensors"
549+
OR path:"open_clip_pytorch_model.bin"
550+
OR path:"pytorch_model.bin"`,
548551
},
549552
paddlenlp: {
550553
prettyLabel: "paddlenlp",

0 commit comments

Comments
 (0)