Skip to content

Commit 065f7f2

Browse files
not-lainWauplin
andauthored
Add pxia (#979)
This pr will support my library **pxia** which can be found in https://github.com/not-lain/pxia the library comes with `AutoModel` support along with 2 architectures for the moment. Let me know if you have any further reviews or feedback. --------- Co-authored-by: Lucain <[email protected]>
1 parent f7cdafe commit 065f7f2

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
@@ -1016,6 +1016,12 @@ export const nemo = (model: ModelData): string[] => {
10161016
return command ?? [`# tag did not correspond to a valid NeMo domain.`];
10171017
};
10181018

1019+
export const pxia = (model: ModelData): string[] => [
1020+
`from pxia import AutoModel
1021+
1022+
model = AutoModel.from_pretrained("${model.id}")`,
1023+
];
1024+
10191025
export const pythae = (model: ModelData): string[] => [
10201026
`from pythae.models import AutoModel
10211027

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)