Skip to content

Commit 930eebd

Browse files
HCookieWauplin
andauthored
Add AnemoI to model-libraries.ts (#1067)
# Add AnemoI https://anemoi-training.readthedocs.io/en/latest/ ECMWF is building a toolkit to enable development of ML weather models. Support will be added for the huggingface hub within the anemoi inference repo. --------- Co-authored-by: Lucain <[email protected]> Co-authored-by: Lucain <[email protected]>
1 parent 3fb4cfb commit 930eebd

File tree

2 files changed

+14
-0
lines changed

2 files changed

+14
-0
lines changed

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

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1118,6 +1118,11 @@ model.set_generation_params(duration=5) # generate 5 seconds.
11181118
descriptions = ['dog barking', 'sirene of an emergency vehicle', 'footsteps in a corridor']
11191119
wav = model.generate(descriptions) # generates 3 samples.`,
11201120
];
1121+
export const anemoi = (model: ModelData): string[] => [
1122+
`from anemoi.inference.runners import DefaultRunner
1123+
runner = DefaultRunner("${model.id}")`,
1124+
];
1125+
11211126

11221127
export const audiocraft = (model: ModelData): string[] => {
11231128
if (model.tags.includes("musicgen")) {

packages/tasks/src/model-libraries.ts

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -75,6 +75,15 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
7575
snippets: snippets.allennlp,
7676
filter: true,
7777
},
78+
anemoi: {
79+
prettyLabel: "AnemoI",
80+
repoName: "AnemoI",
81+
repoUrl: "https://github.com/ecmwf/anemoi-inference",
82+
docsUrl: "https://anemoi-docs.readthedocs.io/en/latest/",
83+
filter: false,
84+
countDownloads: `path_extension:"ckpt"`,
85+
snippets: snippets.anemoi,
86+
},
7887
asteroid: {
7988
prettyLabel: "Asteroid",
8089
repoName: "Asteroid",

0 commit comments

Comments
 (0)