Skip to content

Commit 4e1ae4c

Browse files
Add sam2 to libraries (#831)
Enable stats for sam2 models --------- Co-authored-by: Omar Sanseviero <osanseviero@gmail.com>
1 parent 0e17459 commit 4e1ae4c

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

packages/tasks/src/model-libraries.ts

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -436,6 +436,13 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
436436
snippets: snippets.saelens,
437437
filter: false,
438438
},
439+
sam2: {
440+
prettyLabel: "sam2",
441+
repoName: "sam2",
442+
repoUrl: "https://github.com/facebookresearch/segment-anything-2",
443+
filter: false,
444+
countDownloads: `path_extension:"pt"`,
445+
},
439446
"sample-factory": {
440447
prettyLabel: "sample-factory",
441448
repoName: "sample-factory",

0 commit comments

Comments
 (0)