Skip to content

Commit 5e4beab

Browse files
authored
[YOLOv10] Add safetensors extension (#1213)
As pointed out [here](#1101 (comment)), download counts for YOLOv10 repos which leverage safetensors aren't tracked. This PR fixes that.
1 parent c22681b commit 5e4beab

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
@@ -897,7 +897,7 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
897897
repoName: "YOLOv10",
898898
repoUrl: "https://github.com/THU-MIG/yolov10",
899899
docsUrl: "https://github.com/THU-MIG/yolov10",
900-
countDownloads: `path_extension:"pt"`,
900+
countDownloads: `path_extension:"pt" OR path_extension:"safetensors"`,
901901
snippets: snippets.ultralytics,
902902
},
903903
"3dtopia-xl": {

0 commit comments

Comments
 (0)