Skip to content

Commit 56f465e

Browse files
julian-qWauplin
andauthored
Add countDownloads for Oasis (#1007)
This PR is intended to add download counting support for https://huggingface.co/Etched/oasis-500m. It matches `.pt` files in order to match both the autoencoder and the diffusion model. The relevant github repository (used for inferencing the model) is https://github.com/etched-ai/open-oasis. --------- Co-authored-by: Lucain <[email protected]>
1 parent e57f6ef commit 56f465e

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

packages/tasks/src/model-libraries.ts

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -469,6 +469,12 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
469469
filter: true,
470470
countDownloads: `path_extension:"nemo" OR path:"model_config.yaml"`,
471471
},
472+
open-oasis: {
473+
prettyLabel: "open-oasis",
474+
repoName: "open-oasis",
475+
repoUrl: "https://github.com/etched-ai/open-oasis",
476+
countDownloads: `path:"oasis500m.pt"`,
477+
},
472478
open_clip: {
473479
prettyLabel: "OpenCLIP",
474480
repoName: "OpenCLIP",

0 commit comments

Comments
 (0)