Hugging Face's logo

hf-internal-testing
/
tiny-random-WeatherNext2ForWeatherForecasting

Transformers
Safetensors
weathernext2
tiny-random
testing
Model card Files
xet
 Community
kashif HF Staff commited on
Commit
163414c
·
verified ·
1 Parent(s): 5767866

Tiny random WeatherNext 2 on a 15 degree grid, with real mesh geometry

Browse files
Files changed (3) hide show
  1. config.json +108 -0
  2. model.safetensors +3 -0
  3. preprocessor_config.json +423 -0
config.json ADDED
@@ -0,0 +1,108 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "aggregate_normalization": null,
3
+ "architectures": [
4
+ "WeatherNext2ForWeatherForecasting"
5
+ ],
6
+ "atmospheric_variables": [
7
+ "temperature",
8
+ "geopotential",
9
+ "u_component_of_wind",
10
+ "v_component_of_wind",
11
+ "vertical_velocity",
12
+ "specific_humidity"
13
+ ],
14
+ "attention_bandwidth": 43,
15
+ "attention_dropout": 0.0,
16
+ "attention_k_hop": 2,
17
+ "ball_query_radius_fraction": 0.6,
18
+ "dtype": "float32",
19
+ "edge_hidden_size": 4,
20
+ "forcing_variables": [
21
+ "year_progress_sin",
22
+ "year_progress_cos",
23
+ "day_progress_sin",
24
+ "day_progress_cos"
25
+ ],
26
+ "global_variables": [
27
+ "year_progress_sin",
28
+ "year_progress_cos"
29
+ ],
30
+ "grid_latitudes": 13,
31
+ "grid_longitudes": 24,
32
+ "hidden_act": "gelu_pytorch_tanh",
33
+ "hidden_size": 16,
34
+ "initializer_range": 0.02,
35
+ "input_variables": [
36
+ "temperature",
37
+ "geopotential",
38
+ "u_component_of_wind",
39
+ "v_component_of_wind",
40
+ "vertical_velocity",
41
+ "specific_humidity",
42
+ "2m_temperature",
43
+ "mean_sea_level_pressure",
44
+ "10m_v_component_of_wind",
45
+ "10m_u_component_of_wind",
46
+ "sea_surface_temperature",
47
+ "geopotential_at_surface",
48
+ "land_sea_mask",
49
+ "year_progress_sin",
50
+ "year_progress_cos",
51
+ "day_progress_sin",
52
+ "day_progress_cos"
53
+ ],
54
+ "intermediate_size": 32,
55
+ "layer_norm_eps": 1e-05,
56
+ "mesh_splits": 2,
57
+ "mlp_act": "silu",
58
+ "model_type": "weathernext2",
59
+ "noise_channels": 4,
60
+ "num_attention_heads": 2,
61
+ "num_grid_to_mesh_edges": 576,
62
+ "num_hidden_layers": 2,
63
+ "num_input_timesteps": 2,
64
+ "pressure_levels": [
65
+ 500,
66
+ 850
67
+ ],
68
+ "sigmoid_shifted_outputs": {
69
+ "cyclone_exists_gaussian_unit_mode": 2.0
70
+ },
71
+ "static_variables": [
72
+ "geopotential_at_surface",
73
+ "land_sea_mask"
74
+ ],
75
+ "target_variables": [
76
+ "temperature",
77
+ "geopotential",
78
+ "u_component_of_wind",
79
+ "v_component_of_wind",
80
+ "vertical_velocity",
81
+ "specific_humidity",
82
+ "2m_temperature",
83
+ "mean_sea_level_pressure",
84
+ "10m_v_component_of_wind",
85
+ "10m_u_component_of_wind",
86
+ "sea_surface_temperature",
87
+ "total_precipitation_6hr",
88
+ "cyclone_exists_gaussian_unit_mode",
89
+ "cyclone_all_wind_disc",
90
+ "cyclone_usa_wind_disc",
91
+ "cyclone_usa_r34_ne_radius_disc",
92
+ "cyclone_usa_r34_se_radius_disc",
93
+ "cyclone_usa_r34_sw_radius_disc",
94
+ "cyclone_usa_r34_nw_radius_disc",
95
+ "cyclone_usa_r50_ne_radius_disc",
96
+ "cyclone_usa_r50_se_radius_disc",
97
+ "cyclone_usa_r50_sw_radius_disc",
98
+ "cyclone_usa_r50_nw_radius_disc",
99
+ "cyclone_usa_r64_ne_radius_disc",
100
+ "cyclone_usa_r64_se_radius_disc",
101
+ "cyclone_usa_r64_sw_radius_disc",
102
+ "cyclone_usa_r64_nw_radius_disc",
103
+ "cyclone_usa_rmw_disc",
104
+ "cyclone_usa_pres_disc"
105
+ ],
106
+ "time_step_hours": 6,
107
+ "transformers_version": "5.16.0.dev0"
108
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f778071fa20939493601da744e54c339a0049e1a325c45cbd339a6ff2aea8516
3
+ size 139904
preprocessor_config.json ADDED
@@ -0,0 +1,423 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "atmospheric_variables": [
3
+ "temperature",
4
+ "geopotential",
5
+ "u_component_of_wind",
6
+ "v_component_of_wind",
7
+ "vertical_velocity",
8
+ "specific_humidity"
9
+ ],
10
+ "diffs_stddev_by_level": {
11
+ "10m_u_component_of_wind": [
12
+ 1.0
13
+ ],
14
+ "10m_v_component_of_wind": [
15
+ 1.0
16
+ ],
17
+ "2m_temperature": [
18
+ 1.0
19
+ ],
20
+ "cyclone_all_wind_disc": [
21
+ 1.0
22
+ ],
23
+ "cyclone_exists_gaussian_unit_mode": [
24
+ 1.0
25
+ ],
26
+ "cyclone_usa_pres_disc": [
27
+ 1.0
28
+ ],
29
+ "cyclone_usa_r34_ne_radius_disc": [
30
+ 1.0
31
+ ],
32
+ "cyclone_usa_r34_nw_radius_disc": [
33
+ 1.0
34
+ ],
35
+ "cyclone_usa_r34_se_radius_disc": [
36
+ 1.0
37
+ ],
38
+ "cyclone_usa_r34_sw_radius_disc": [
39
+ 1.0
40
+ ],
41
+ "cyclone_usa_r50_ne_radius_disc": [
42
+ 1.0
43
+ ],
44
+ "cyclone_usa_r50_nw_radius_disc": [
45
+ 1.0
46
+ ],
47
+ "cyclone_usa_r50_se_radius_disc": [
48
+ 1.0
49
+ ],
50
+ "cyclone_usa_r50_sw_radius_disc": [
51
+ 1.0
52
+ ],
53
+ "cyclone_usa_r64_ne_radius_disc": [
54
+ 1.0
55
+ ],
56
+ "cyclone_usa_r64_nw_radius_disc": [
57
+ 1.0
58
+ ],
59
+ "cyclone_usa_r64_se_radius_disc": [
60
+ 1.0
61
+ ],
62
+ "cyclone_usa_r64_sw_radius_disc": [
63
+ 1.0
64
+ ],
65
+ "cyclone_usa_rmw_disc": [
66
+ 1.0
67
+ ],
68
+ "cyclone_usa_wind_disc": [
69
+ 1.0
70
+ ],
71
+ "day_progress_cos": [
72
+ 1.0
73
+ ],
74
+ "day_progress_sin": [
75
+ 1.0
76
+ ],
77
+ "geopotential": [
78
+ 1.0,
79
+ 1.0
80
+ ],
81
+ "geopotential_at_surface": [
82
+ 1.0
83
+ ],
84
+ "land_sea_mask": [
85
+ 1.0
86
+ ],
87
+ "mean_sea_level_pressure": [
88
+ 1.0
89
+ ],
90
+ "sea_surface_temperature": [
91
+ 1.0
92
+ ],
93
+ "specific_humidity": [
94
+ 1.0,
95
+ 1.0
96
+ ],
97
+ "temperature": [
98
+ 1.0,
99
+ 1.0
100
+ ],
101
+ "total_precipitation_6hr": [
102
+ 1.0
103
+ ],
104
+ "u_component_of_wind": [
105
+ 1.0,
106
+ 1.0
107
+ ],
108
+ "v_component_of_wind": [
109
+ 1.0,
110
+ 1.0
111
+ ],
112
+ "vertical_velocity": [
113
+ 1.0,
114
+ 1.0
115
+ ],
116
+ "year_progress_cos": [
117
+ 1.0
118
+ ],
119
+ "year_progress_sin": [
120
+ 1.0
121
+ ]
122
+ },
123
+ "feature_extractor_type": "WeatherNext2FeatureExtractor",
124
+ "forcing_variables": [
125
+ "year_progress_sin",
126
+ "year_progress_cos",
127
+ "day_progress_sin",
128
+ "day_progress_cos"
129
+ ],
130
+ "global_variables": [
131
+ "year_progress_sin",
132
+ "year_progress_cos"
133
+ ],
134
+ "grid_latitudes": 13,
135
+ "grid_longitudes": 24,
136
+ "input_variables": [
137
+ "temperature",
138
+ "geopotential",
139
+ "u_component_of_wind",
140
+ "v_component_of_wind",
141
+ "vertical_velocity",
142
+ "specific_humidity",
143
+ "2m_temperature",
144
+ "mean_sea_level_pressure",
145
+ "10m_v_component_of_wind",
146
+ "10m_u_component_of_wind",
147
+ "sea_surface_temperature",
148
+ "geopotential_at_surface",
149
+ "land_sea_mask",
150
+ "year_progress_sin",
151
+ "year_progress_cos",
152
+ "day_progress_sin",
153
+ "day_progress_cos"
154
+ ],
155
+ "mean_by_level": {
156
+ "10m_u_component_of_wind": [
157
+ 0.0
158
+ ],
159
+ "10m_v_component_of_wind": [
160
+ 0.0
161
+ ],
162
+ "2m_temperature": [
163
+ 0.0
164
+ ],
165
+ "cyclone_all_wind_disc": [
166
+ 0.0
167
+ ],
168
+ "cyclone_exists_gaussian_unit_mode": [
169
+ 0.0
170
+ ],
171
+ "cyclone_usa_pres_disc": [
172
+ 0.0
173
+ ],
174
+ "cyclone_usa_r34_ne_radius_disc": [
175
+ 0.0
176
+ ],
177
+ "cyclone_usa_r34_nw_radius_disc": [
178
+ 0.0
179
+ ],
180
+ "cyclone_usa_r34_se_radius_disc": [
181
+ 0.0
182
+ ],
183
+ "cyclone_usa_r34_sw_radius_disc": [
184
+ 0.0
185
+ ],
186
+ "cyclone_usa_r50_ne_radius_disc": [
187
+ 0.0
188
+ ],
189
+ "cyclone_usa_r50_nw_radius_disc": [
190
+ 0.0
191
+ ],
192
+ "cyclone_usa_r50_se_radius_disc": [
193
+ 0.0
194
+ ],
195
+ "cyclone_usa_r50_sw_radius_disc": [
196
+ 0.0
197
+ ],
198
+ "cyclone_usa_r64_ne_radius_disc": [
199
+ 0.0
200
+ ],
201
+ "cyclone_usa_r64_nw_radius_disc": [
202
+ 0.0
203
+ ],
204
+ "cyclone_usa_r64_se_radius_disc": [
205
+ 0.0
206
+ ],
207
+ "cyclone_usa_r64_sw_radius_disc": [
208
+ 0.0
209
+ ],
210
+ "cyclone_usa_rmw_disc": [
211
+ 0.0
212
+ ],
213
+ "cyclone_usa_wind_disc": [
214
+ 0.0
215
+ ],
216
+ "day_progress_cos": [
217
+ 0.0
218
+ ],
219
+ "day_progress_sin": [
220
+ 0.0
221
+ ],
222
+ "geopotential": [
223
+ 0.0,
224
+ 0.0
225
+ ],
226
+ "geopotential_at_surface": [
227
+ 0.0
228
+ ],
229
+ "land_sea_mask": [
230
+ 0.0
231
+ ],
232
+ "mean_sea_level_pressure": [
233
+ 0.0
234
+ ],
235
+ "sea_surface_temperature": [
236
+ 0.0
237
+ ],
238
+ "specific_humidity": [
239
+ 0.0,
240
+ 0.0
241
+ ],
242
+ "temperature": [
243
+ 0.0,
244
+ 0.0
245
+ ],
246
+ "total_precipitation_6hr": [
247
+ 0.0
248
+ ],
249
+ "u_component_of_wind": [
250
+ 0.0,
251
+ 0.0
252
+ ],
253
+ "v_component_of_wind": [
254
+ 0.0,
255
+ 0.0
256
+ ],
257
+ "vertical_velocity": [
258
+ 0.0,
259
+ 0.0
260
+ ],
261
+ "year_progress_cos": [
262
+ 0.0
263
+ ],
264
+ "year_progress_sin": [
265
+ 0.0
266
+ ]
267
+ },
268
+ "nan_fill_values": {},
269
+ "num_input_timesteps": 2,
270
+ "pressure_levels": [
271
+ 500,
272
+ 850
273
+ ],
274
+ "static_variables": [
275
+ "geopotential_at_surface",
276
+ "land_sea_mask"
277
+ ],
278
+ "stddev_by_level": {
279
+ "10m_u_component_of_wind": [
280
+ 1.0
281
+ ],
282
+ "10m_v_component_of_wind": [
283
+ 1.0
284
+ ],
285
+ "2m_temperature": [
286
+ 1.0
287
+ ],
288
+ "cyclone_all_wind_disc": [
289
+ 1.0
290
+ ],
291
+ "cyclone_exists_gaussian_unit_mode": [
292
+ 1.0
293
+ ],
294
+ "cyclone_usa_pres_disc": [
295
+ 1.0
296
+ ],
297
+ "cyclone_usa_r34_ne_radius_disc": [
298
+ 1.0
299
+ ],
300
+ "cyclone_usa_r34_nw_radius_disc": [
301
+ 1.0
302
+ ],
303
+ "cyclone_usa_r34_se_radius_disc": [
304
+ 1.0
305
+ ],
306
+ "cyclone_usa_r34_sw_radius_disc": [
307
+ 1.0
308
+ ],
309
+ "cyclone_usa_r50_ne_radius_disc": [
310
+ 1.0
311
+ ],
312
+ "cyclone_usa_r50_nw_radius_disc": [
313
+ 1.0
314
+ ],
315
+ "cyclone_usa_r50_se_radius_disc": [
316
+ 1.0
317
+ ],
318
+ "cyclone_usa_r50_sw_radius_disc": [
319
+ 1.0
320
+ ],
321
+ "cyclone_usa_r64_ne_radius_disc": [
322
+ 1.0
323
+ ],
324
+ "cyclone_usa_r64_nw_radius_disc": [
325
+ 1.0
326
+ ],
327
+ "cyclone_usa_r64_se_radius_disc": [
328
+ 1.0
329
+ ],
330
+ "cyclone_usa_r64_sw_radius_disc": [
331
+ 1.0
332
+ ],
333
+ "cyclone_usa_rmw_disc": [
334
+ 1.0
335
+ ],
336
+ "cyclone_usa_wind_disc": [
337
+ 1.0
338
+ ],
339
+ "day_progress_cos": [
340
+ 1.0
341
+ ],
342
+ "day_progress_sin": [
343
+ 1.0
344
+ ],
345
+ "geopotential": [
346
+ 1.0,
347
+ 1.0
348
+ ],
349
+ "geopotential_at_surface": [
350
+ 1.0
351
+ ],
352
+ "land_sea_mask": [
353
+ 1.0
354
+ ],
355
+ "mean_sea_level_pressure": [
356
+ 1.0
357
+ ],
358
+ "sea_surface_temperature": [
359
+ 1.0
360
+ ],
361
+ "specific_humidity": [
362
+ 1.0,
363
+ 1.0
364
+ ],
365
+ "temperature": [
366
+ 1.0,
367
+ 1.0
368
+ ],
369
+ "total_precipitation_6hr": [
370
+ 1.0
371
+ ],
372
+ "u_component_of_wind": [
373
+ 1.0,
374
+ 1.0
375
+ ],
376
+ "v_component_of_wind": [
377
+ 1.0,
378
+ 1.0
379
+ ],
380
+ "vertical_velocity": [
381
+ 1.0,
382
+ 1.0
383
+ ],
384
+ "year_progress_cos": [
385
+ 1.0
386
+ ],
387
+ "year_progress_sin": [
388
+ 1.0
389
+ ]
390
+ },
391
+ "target_variables": [
392
+ "temperature",
393
+ "geopotential",
394
+ "u_component_of_wind",
395
+ "v_component_of_wind",
396
+ "vertical_velocity",
397
+ "specific_humidity",
398
+ "2m_temperature",
399
+ "mean_sea_level_pressure",
400
+ "10m_v_component_of_wind",
401
+ "10m_u_component_of_wind",
402
+ "sea_surface_temperature",
403
+ "total_precipitation_6hr",
404
+ "cyclone_exists_gaussian_unit_mode",
405
+ "cyclone_all_wind_disc",
406
+ "cyclone_usa_wind_disc",
407
+ "cyclone_usa_r34_ne_radius_disc",
408
+ "cyclone_usa_r34_se_radius_disc",
409
+ "cyclone_usa_r34_sw_radius_disc",
410
+ "cyclone_usa_r34_nw_radius_disc",
411
+ "cyclone_usa_r50_ne_radius_disc",
412
+ "cyclone_usa_r50_se_radius_disc",
413
+ "cyclone_usa_r50_sw_radius_disc",
414
+ "cyclone_usa_r50_nw_radius_disc",
415
+ "cyclone_usa_r64_ne_radius_disc",
416
+ "cyclone_usa_r64_se_radius_disc",
417
+ "cyclone_usa_r64_sw_radius_disc",
418
+ "cyclone_usa_r64_nw_radius_disc",
419
+ "cyclone_usa_rmw_disc",
420
+ "cyclone_usa_pres_disc"
421
+ ],
422
+ "time_step_hours": 6
423
+ }