Skip to content

Commit 6ff6584

Browse files
authored
Rework anemoi snippet (huggingface#1071)
# Rework anemoi snippet - Fix snippet to allow for easy instantiation See ecmwf/anemoi-inference#77 ```python from anemoi.inference.runners.default import DefaultRunner from anemoi.inference.config import Configuration # Create Configuration config = Configuration(checkpoint = {"huggingface":{"repo_id":"ecwmf/aifs-single"}}) # Load Runner runner = DefaultRunner(config) ```
1 parent c0b2ca3 commit 6ff6584

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

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

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1119,8 +1119,12 @@ descriptions = ['dog barking', 'sirene of an emergency vehicle', 'footsteps in a
11191119
wav = model.generate(descriptions) # generates 3 samples.`,
11201120
];
11211121
export const anemoi = (model: ModelData): string[] => [
1122-
`from anemoi.inference.runners import DefaultRunner
1123-
runner = DefaultRunner("${model.id}")`,
1122+
`from anemoi.inference.runners.default import DefaultRunner
1123+
from anemoi.inference.config import Configuration
1124+
# Create Configuration
1125+
config = Configuration(checkpoint = {"huggingface":{"repo_id":"${model.id}"}})
1126+
# Load Runner
1127+
runner = DefaultRunner(config)`,
11241128
];
11251129

11261130

0 commit comments

Comments
 (0)