add path ext to yolov8 and yolov10

huggingface · Jan 9, 2025 · e32f38d · e32f38d
1 parent 5cdf3f2
commit e32f38d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/packages/tasks/src/model-libraries.ts b/packages/tasks/src/model-libraries.ts
@@ -849,13 +849,15 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
 		repoUrl: "https://github.com/ultralytics/ultralytics",
 		docsUrl: "https://github.com/ultralytics/ultralytics",
 		filter: false,
+		countDownloads: `path_extension:"pt"`,
 		snippets: snippets.yolov8,
 	},
 	yolov10: {
 		prettyLabel: "YOLOv10",
 		repoName: "yolov10",
 		repoUrl: "https://github.com/THU-MIG/yolov10",
 		docsUrl: "https://github.com/THU-MIG/yolov10",
+		countDownloads: `path_extension:"pt"`,
 		snippets: snippets.yolov10,
 	},
 	whisperkit: {