latterworks commited on
Commit
d854254
·
verified ·
1 Parent(s): 690fda8

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +32 -0
README.md ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - terrain-generation
4
+ - satclip
5
+ - diffusion
6
+ - heightmap
7
+ - procedural
8
+ license: mit
9
+ ---
10
+
11
+ # SatCLIP Terrain Generator
12
+
13
+ Generate realistic terrain heightmaps conditioned on geographic coordinates via SatCLIP embeddings.
14
+
15
+ ## Usage
16
+ ```python
17
+ from huggingface_hub import hf_hub_download
18
+ import torch
19
+
20
+ ckpt = hf_hub_download("latterworks/satclip-terrain", "model.pt")
21
+ weights = torch.load(ckpt)
22
+
23
+ # Requires satclip + model architecture from training code
24
+ model.vae.load_state_dict(weights['vae'])
25
+ model.prior.load_state_dict(weights['prior'])
26
+ model.refiner.load_state_dict(weights['refiner'])
27
+ ```
28
+
29
+ ## Training
30
+ - VAE + diffusion prior on SRTM 90m elevation data
31
+ - Colorado/Utah region tiles
32
+ - Conditioned on microsoft/SatCLIP-ResNet18-L40 location embeddings