Skip to content

Commit d92f87c

Browse files
committed
add pxia code snippet
1 parent 78098cc commit d92f87c

File tree

2 files changed

+17
-0
lines changed

2 files changed

+17
-0
lines changed

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

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -997,6 +997,16 @@ export const nemo = (model: ModelData): string[] => {
997997
return command ?? [`# tag did not correspond to a valid NeMo domain.`];
998998
};
999999

1000+
export const pxia = (model: ModelData): string[] => [
1001+
`# Installation
1002+
1003+
pip install pxia
1004+
`,
1005+
`# Load the model
1006+
from pxia import AutoModel
1007+
model = AutoModel.from_pretrained("${model.id}")`,
1008+
];
1009+
10001010
export const pythae = (model: ModelData): string[] => [
10011011
`from pythae.models import AutoModel
10021012

packages/tasks/src/model-libraries.ts

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -481,6 +481,13 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
481481
filter: true,
482482
countDownloads: `path:"adapter_config.json"`,
483483
},
484+
pxia: {
485+
prettyLabel: "pxia",
486+
repoName: "pxia",
487+
repoUrl: "https://github.com/not-lain/pxia",
488+
snippets: snippets.pxia,
489+
filter: false,
490+
},
484491
"pyannote-audio": {
485492
prettyLabel: "pyannote.audio",
486493
repoName: "pyannote-audio",

0 commit comments

Comments
 (0)