Views
No views yet
1from huggingface_hub import hf_hub_download
2import torch
3
4ckpt = hf_hub_download("latterworks/satclip-terrain", "model.pt")
5weights = torch.load(ckpt)
6
7# Requires satclip + model architecture from training code
8model.vae.load_state_dict(weights['vae'])
9model.prior.load_state_dict(weights['prior'])
10model.refiner.load_state_dict(weights['refiner'])