Hugging Face's logo

xandergos
/
terrain-diffusion-90m-pool2

Diffusers
Safetensors
Model card Files
xet
Community
xandergos commited on
Commit
f657baf
·
verified ·
1 Parent(s): 889dbf6

Initial commit

Browse files
Files changed (7) hide show
  1. base_model/config.json +39 -0
  2. base_model/diffusion_pytorch_model.safetensors +3 -0
  3. coarse_model/config.json +51 -0
  4. coarse_model/diffusion_pytorch_model.safetensors +3 -0
  5. config.json +27 -0
  6. decoder_model/config.json +28 -0
  7. decoder_model/diffusion_pytorch_model.safetensors +3 -0
base_model/config.json ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "EDMUnet2D",
3
+ "_diffusers_version": "0.30.3",
4
+ "_name_or_path": "./checkpoints/models/consistency_base-192x3/",
5
+ "attn_resolutions": [
6
+ 8,
7
+ 16
8
+ ],
9
+ "block_kwargs": {
10
+ "dropout": 0.1
11
+ },
12
+ "concat_balance": 0.5,
13
+ "conditional_inputs": [
14
+ [
15
+ "tensor",
16
+ 58,
17
+ 1.0
18
+ ]
19
+ ],
20
+ "disable_out_gain": false,
21
+ "emb_channels": null,
22
+ "encode_only": false,
23
+ "fourier_scale": "pos",
24
+ "image_size": 512,
25
+ "in_channels": 5,
26
+ "layers_per_block": 3,
27
+ "logvar_channels": 128,
28
+ "midblock_attention": true,
29
+ "model_channel_mults": [
30
+ 1,
31
+ 2,
32
+ 3,
33
+ 4
34
+ ],
35
+ "model_channels": 192,
36
+ "n_logvar": 1,
37
+ "noise_emb_dims": null,
38
+ "out_channels": 5
39
+ }
base_model/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e426277db86517335d4b0bc02b3d456bb812b2a8726ea010635686f77373be36
3
+ size 1014772076
coarse_model/config.json ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "EDMUnet2D",
3
+ "_diffusers_version": "0.30.3",
4
+ "_name_or_path": "./checkpoints/models/diffusion_coarse/",
5
+ "attn_resolutions": [],
6
+ "block_kwargs": {},
7
+ "concat_balance": 0.5,
8
+ "conditional_inputs": [
9
+ [
10
+ "float",
11
+ 64,
12
+ 0.2
13
+ ],
14
+ [
15
+ "float",
16
+ 64,
17
+ 0.2
18
+ ],
19
+ [
20
+ "float",
21
+ 64,
22
+ 0.2
23
+ ],
24
+ [
25
+ "float",
26
+ 64,
27
+ 0.2
28
+ ],
29
+ [
30
+ "float",
31
+ 64,
32
+ 0.2
33
+ ]
34
+ ],
35
+ "disable_out_gain": false,
36
+ "emb_channels": null,
37
+ "encode_only": false,
38
+ "fourier_scale": "pos",
39
+ "image_size": 16,
40
+ "in_channels": 11,
41
+ "layers_per_block": 2,
42
+ "logvar_channels": 128,
43
+ "midblock_attention": false,
44
+ "model_channel_mults": [
45
+ 1
46
+ ],
47
+ "model_channels": 128,
48
+ "n_logvar": 1,
49
+ "noise_emb_dims": null,
50
+ "out_channels": 6
51
+ }
coarse_model/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:13c21db4581d2072db56fd76fe3b92fa2d5efa8805be2aae0b63448b5d28ac5c
3
+ size 11200936
config.json ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "WorldPipeline",
3
+ "_diffusers_version": "0.30.3",
4
+ "coarse_pooling": 2,
5
+ "cond_snr": [
6
+ 0.5,
7
+ 0.5,
8
+ 0.5,
9
+ 0.5,
10
+ 0.5
11
+ ],
12
+ "drop_water_pct": 0.5,
13
+ "elev_coarse_pool_mode": "avg",
14
+ "frequency_mult": [
15
+ 1,
16
+ 1,
17
+ 1,
18
+ 1,
19
+ 1
20
+ ],
21
+ "histogram_raw": null,
22
+ "latent_compression": 8,
23
+ "native_resolution": 90,
24
+ "p5_coarse_pool_mode": "avg",
25
+ "residual_mean": 0.0,
26
+ "residual_std": 1.1678
27
+ }
decoder_model/config.json ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "EDMUnet2D",
3
+ "_diffusers_version": "0.30.3",
4
+ "_name_or_path": "./checkpoints/models/consistency_decoder-64x3/",
5
+ "attn_resolutions": [],
6
+ "block_kwargs": null,
7
+ "concat_balance": 0.5,
8
+ "conditional_inputs": [],
9
+ "disable_out_gain": false,
10
+ "emb_channels": null,
11
+ "encode_only": false,
12
+ "fourier_scale": "pos",
13
+ "image_size": 512,
14
+ "in_channels": 5,
15
+ "layers_per_block": 3,
16
+ "logvar_channels": 128,
17
+ "midblock_attention": false,
18
+ "model_channel_mults": [
19
+ 1,
20
+ 2,
21
+ 3,
22
+ 4
23
+ ],
24
+ "model_channels": 64,
25
+ "n_logvar": 1,
26
+ "noise_emb_dims": null,
27
+ "out_channels": 1
28
+ }
decoder_model/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b6c7fa99f836ad75c514236c9529e18a68ea207ed59dd39fd1341fc9a8a03bcc
3
+ size 111709108