Hugging Face's logo

mally-2000
/
saii-cldm-synthetic

Image-to-Image
Diffusers
Safetensors
SeismicImpInvCLDMPipeline
seismic-inversion
impedance-inversion
diffusion
ddpm
cldm
overthrust
synthetic-data
Model card Files
xet
 Community
mally-2000 commited on
Commit
bd47be2
·
verified ·
1 Parent(s): 0f3b084

Upload Seismic LDDPM Pipeline

Browse files
Files changed (8) hide show
  1. condition_encoder/config.json +10 -0
  2. condition_encoder/diffusion_pytorch_model.safetensors +3 -0
  3. model_index.json +20 -0
  4. scheduler/scheduler_config.json +19 -0
  5. unet/config.json +46 -0
  6. unet/diffusion_pytorch_model.safetensors +3 -0
  7. vq_model/config.json +34 -0
  8. vq_model/diffusion_pytorch_model.safetensors +3 -0
condition_encoder/config.json ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "SeismicWaveletEncoder",
3
+ "_diffusers_version": "0.30.3",
4
+ "_name_or_path": "/home/coco/zhixingyun/test/cldm2/converted_encoder_wrong",
5
+ "channels": 4,
6
+ "in_channels": 1,
7
+ "mode": "reflect",
8
+ "out_channels": 4,
9
+ "wavelet_type": "haar"
10
+ }
condition_encoder/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:89c6580939d565cceccb89d5f4d1482990cc2e4f4ecb0140aa5a92c6a0e294a9
3
+ size 2224
model_index.json ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "SeismicImpInvLDDPMPipeline",
3
+ "_diffusers_version": "0.30.3",
4
+ "condition_encoder": [
5
+ "ldm.modules.encoders.seismic_wavelet_encoder",
6
+ "SeismicWaveletEncoder"
7
+ ],
8
+ "scheduler": [
9
+ "diffusers",
10
+ "DDPMScheduler"
11
+ ],
12
+ "unet": [
13
+ "diffusers",
14
+ "UNet2DModel"
15
+ ],
16
+ "vq_model": [
17
+ "diffusers",
18
+ "VQModel"
19
+ ]
20
+ }
scheduler/scheduler_config.json ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "DDPMScheduler",
3
+ "_diffusers_version": "0.30.3",
4
+ "beta_end": 0.0195,
5
+ "beta_schedule": "linear",
6
+ "beta_start": 0.0015,
7
+ "clip_sample": false,
8
+ "clip_sample_range": 1.0,
9
+ "dynamic_thresholding_ratio": 0.995,
10
+ "num_train_timesteps": 1000,
11
+ "prediction_type": "epsilon",
12
+ "rescale_betas_zero_snr": false,
13
+ "sample_max_value": 1.0,
14
+ "steps_offset": 0,
15
+ "thresholding": false,
16
+ "timestep_spacing": "leading",
17
+ "trained_betas": null,
18
+ "variance_type": "fixed_small"
19
+ }
unet/config.json ADDED
@@ -0,0 +1,46 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DModel",
3
+ "_diffusers_version": "0.30.3",
4
+ "_name_or_path": "/home/coco/zhixingyun/test/cldm2/tmp/a101_diffusers_unet",
5
+ "act_fn": "silu",
6
+ "add_attention": true,
7
+ "attention_head_dim": 32,
8
+ "attn_norm_num_groups": null,
9
+ "block_out_channels": [
10
+ 224,
11
+ 448,
12
+ 672,
13
+ 896
14
+ ],
15
+ "center_input_sample": false,
16
+ "class_embed_type": null,
17
+ "down_block_types": [
18
+ "DownBlock2D",
19
+ "AttnDownBlock2D",
20
+ "AttnDownBlock2D",
21
+ "AttnDownBlock2D"
22
+ ],
23
+ "downsample_padding": 1,
24
+ "downsample_type": "conv",
25
+ "dropout": 0.0,
26
+ "flip_sin_to_cos": true,
27
+ "freq_shift": 0,
28
+ "in_channels": 6,
29
+ "layers_per_block": 2,
30
+ "mid_block_scale_factor": 1,
31
+ "norm_eps": 1e-05,
32
+ "norm_num_groups": 32,
33
+ "num_class_embeds": null,
34
+ "num_train_timesteps": null,
35
+ "out_channels": 1,
36
+ "resnet_time_scale_shift": "default",
37
+ "sample_size": 64,
38
+ "time_embedding_type": "positional",
39
+ "up_block_types": [
40
+ "AttnUpBlock2D",
41
+ "AttnUpBlock2D",
42
+ "AttnUpBlock2D",
43
+ "UpBlock2D"
44
+ ],
45
+ "upsample_type": "conv"
46
+ }
unet/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b320ff97f024c8668327f17029a41f3a8935043549c502f8a50bc3f65e1b48f0
3
+ size 1096279404
vq_model/config.json ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "VQModel",
3
+ "_diffusers_version": "0.30.3",
4
+ "_name_or_path": "/home/coco/zhixingyun/use_model_param/old_vqgan_diffusers_vqmodel",
5
+ "act_fn": "silu",
6
+ "block_out_channels": [
7
+ 128,
8
+ 256,
9
+ 512
10
+ ],
11
+ "down_block_types": [
12
+ "DownEncoderBlock2D",
13
+ "DownEncoderBlock2D",
14
+ "DownEncoderBlock2D"
15
+ ],
16
+ "force_upcast": false,
17
+ "in_channels": 1,
18
+ "latent_channels": 3,
19
+ "layers_per_block": 2,
20
+ "lookup_from_codebook": false,
21
+ "mid_block_add_attention": true,
22
+ "norm_num_groups": 32,
23
+ "norm_type": "group",
24
+ "num_vq_embeddings": 32768,
25
+ "out_channels": 1,
26
+ "sample_size": 256,
27
+ "scaling_factor": 1.0,
28
+ "up_block_types": [
29
+ "UpDecoderBlock2D",
30
+ "UpDecoderBlock2D",
31
+ "UpDecoderBlock2D"
32
+ ],
33
+ "vq_embed_dim": 1
34
+ }
vq_model/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a526a35bf564aca5084749e7ca361d7df9ea2b6487321cf1207b06735d046a47
3
+ size 221328288