Skip to content

Commit 58d5367

Browse files
authored
add araclip (#1264)
this pr will add araclip to the list of supported libraries * model : https://huggingface.co/Arabic-Clip/araclip * repo : https://github.com/Arabic-Clip/Araclip
1 parent b5ce53c commit 58d5367

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
@@ -46,6 +46,12 @@ export const allennlp = (model: ModelData): string[] => {
4646
return allennlpUnknown(model);
4747
};
4848

49+
export const araclip = (model: ModelData): string[] => [
50+
`from araclip import AraClip
51+
52+
model = AraClip.from_pretrained("${model.id}")`,
53+
];
54+
4955
export const asteroid = (model: ModelData): string[] => [
5056
`from asteroid.models import BaseModel
5157

packages/tasks/src/model-libraries.ts

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -84,6 +84,13 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
8484
countDownloads: `path_extension:"ckpt"`,
8585
snippets: snippets.anemoi,
8686
},
87+
araclip: {
88+
prettyLabel: "AraClip",
89+
repoName: "AraClip",
90+
repoUrl: "https://huggingface.co/Arabic-Clip/araclip",
91+
filter: false,
92+
snippets: snippets.araclip,
93+
},
8794
asteroid: {
8895
prettyLabel: "Asteroid",
8996
repoName: "Asteroid",

0 commit comments

Comments
 (0)