Views
No views yet
1import torch
2from pw_foucast import PW_FouCast
3from safetensors.torch import load_model
4from huggingface_hub import hf_hub_download
5
6MODEL_REGISTRY = {
7 'pw_foucast': PW_FouCast,
8}
9ModelClass = MODEL_REGISTRY.get(args.model.lower())
10
11model = ModelClass(**model_kwargs).to(args.device)
12model = torch.nn.DataParallel(model)
13
14# Load the model from Hugging Face
15weights_path = hf_hub_download(repo_id=f"Onemiss/PW-FouCast", filename=f"{args.model}/{args.dataset}/model.safetensors")
16load_model(model, weights_path)
17
18# Eval
19model.eval()
20……1@article{qin2026extending,
2 title={Extending Precipitation Nowcasting Horizons via Spectral Fusion of Radar Observations and Foundation Model Priors},
3 author={Yuze Qin, Qingyong Li, Zhiqing Guo, Wen Wang, Yan Liu, Yangli-ao Geng},
4 journal={arXiv preprint arXiv:2603.21768},
5 year={2026}
6}