tarekziade HF Staff
Consolidate MIC usage, slim weights and reviewed framework expectations
3ce5455 verified | { | |
| "mic": "0.3", | |
| "name": "WeatherNext2 forecast", | |
| "usage": "Feature extraction, a batched forecast, physical postprocessing, and ensemble CRPS training", | |
| "architecture": "WeatherNext2ForWeatherForecasting", | |
| "checkpoints": { | |
| "kashif/weathernext2-mini": { | |
| "fixture": "huggingface-mic/weathernext2-mic" | |
| } | |
| }, | |
| "fixtures": { | |
| "huggingface-mic/weathernext2-mic": { | |
| "built_from": "kashif/weathernext2-mini", | |
| "covers": [ | |
| "kashif/weathernext2-mini" | |
| ], | |
| "build": { | |
| "seed": 17, | |
| "model_class": "WeatherNext2ForWeatherForecasting", | |
| "config_overrides": { | |
| "fields": { | |
| "graph_config.edge_hidden_size": 64, | |
| "grid_latitudes": 9, | |
| "grid_longitudes": 16, | |
| "mesh_splits": 2, | |
| "attention_k_hop": 2 | |
| }, | |
| "reason": "Shrink the network, weather grid and mesh together; regenerate genuine spherical geometry and its mask." | |
| } | |
| } | |
| } | |
| }, | |
| "integrations": { | |
| "transformers": { | |
| "entry_point": "integrations/transformers/usage.md", | |
| "format": "runnable-markdown", | |
| "requires": [ | |
| "transformers>=5", | |
| "torch" | |
| ], | |
| "targets": [ | |
| "fixtures" | |
| ], | |
| "variants": { | |
| "default": {}, | |
| "eager": { | |
| "attn_implementation": "eager" | |
| }, | |
| "bfloat16": { | |
| "dtype": "bfloat16" | |
| } | |
| }, | |
| "training": true | |
| } | |
| }, | |
| "outputs": { | |
| "runtime": { | |
| "compare": "exact" | |
| }, | |
| "shape": { | |
| "compare": "exact" | |
| }, | |
| "prediction": { | |
| "compare": "allclose", | |
| "atol": 0.0001, | |
| "rtol": 0.0001 | |
| }, | |
| "temperature": { | |
| "compare": "allclose", | |
| "atol": 0.0001, | |
| "rtol": 0.0001 | |
| }, | |
| "finite": { | |
| "compare": "true" | |
| }, | |
| "training.loss": { | |
| "compare": "allclose", | |
| "atol": 0.0001, | |
| "rtol": 0.0001 | |
| }, | |
| "training.gradients": { | |
| "compare": "all_true" | |
| }, | |
| "training.updated": { | |
| "compare": "true" | |
| } | |
| } | |
| } | |