specific_humidity
temperature
velocitiy_u
velocity_v
velocity_z

model_id: id1xn47ihk, from training 11uq55oh , epoch 40
token_size: [3, 9, 9]
num_tokens: [12, 6, 12]
ens_dim: 16

specific_humidity id1xn47ihk [3, 9, 9] [12, 6, 12] 16