Skip to content

Commit b7d2590

Browse files
authored
[integration] Add hezar to libraries and snippets (#934)
This PR adds integration for `hezar` mainly for download stats and supporting code snippets for the models built with this library. This library currently has 30 models and 10 datasets with 70K installs and counting. This library has been around for a year now and since our users requested this a lot, we decided to add this integration for convenience. Library repo: https://github.com/hezarai/hezar Creator: @arxyzan Let me know if this PR conforms to the necessary guidelines. Thank you guys ;)
1 parent 1a6da30 commit b7d2590

File tree

2 files changed

+13
-0
lines changed

2 files changed

+13
-0
lines changed

packages/tasks/src/model-libraries-snippets.ts

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1032,4 +1032,10 @@ export const threedtopia_xl = (model: ModelData): string[] => [
10321032
model = threedtopia_xl.from_pretrained("${model.id}")
10331033
model.generate(cond="path/to/image.png")`,
10341034
];
1035+
1036+
export const hezar = (model: ModelData): string[] => [
1037+
`from hezar import Model
1038+
1039+
model = Model.load("${model.id}")`,
1040+
];
10351041
//#endregion

packages/tasks/src/model-libraries.ts

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -295,6 +295,13 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
295295
repoUrl: "https://github.com/fudan-generative-vision/hallo",
296296
countDownloads: `path:"hallo/net.pth"`,
297297
},
298+
hezar: {
299+
prettyLabel: "Hezar",
300+
repoName: "Hezar",
301+
repoUrl: "https://github.com/hezarai/hezar",
302+
docsUrl: "https://hezarai.github.io/hezar",
303+
countDownloads: `path:"model_config.yaml" OR path:"embedding/embedding_config.yaml"`,
304+
},
298305
"hunyuan-dit": {
299306
prettyLabel: "HunyuanDiT",
300307
repoName: "HunyuanDiT",

0 commit comments

Comments
 (0)