Add generative model checkpoints for Mask generation (DDPM) and CMRI generation (DDPM, LDM, FM)

ddpm_model/model_index.json

@@ -0,0 +1,12 @@
+{
+  "_class_name": "SegGuidedDDPMPipeline",
+  "_diffusers_version": "0.35.2",
+  "scheduler": [
+    "diffusers",
+    "DDPMScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

ddpm_model/scheduler/scheduler_config.json

@@ -0,0 +1,19 @@
+{
+  "_class_name": "DDPMScheduler",
+  "_diffusers_version": "0.35.2",
+  "beta_end": 0.02,
+  "beta_schedule": "linear",
+  "beta_start": 0.0001,
+  "clip_sample": true,
+  "clip_sample_range": 1.0,
+  "dynamic_thresholding_ratio": 0.995,
+  "num_train_timesteps": 1000,
+  "prediction_type": "epsilon",
+  "rescale_betas_zero_snr": false,
+  "sample_max_value": 1.0,
+  "steps_offset": 0,
+  "thresholding": false,
+  "timestep_spacing": "leading",
+  "trained_betas": null,
+  "variance_type": "fixed_small"
+}

ddpm_model/unet/config.json

@@ -0,0 +1,54 @@
+{
+  "_class_name": "UNet2DModel",
+  "_diffusers_version": "0.35.2",
+  "_name_or_path": "ddpm-150-model-v2/unet",
+  "act_fn": "silu",
+  "add_attention": true,
+  "attention_head_dim": 8,
+  "attn_norm_num_groups": null,
+  "block_out_channels": [
+    128,
+    128,
+    256,
+    256,
+    512,
+    512
+  ],
+  "center_input_sample": false,
+  "class_embed_type": null,
+  "down_block_types": [
+    "DownBlock2D",
+    "DownBlock2D",
+    "DownBlock2D",
+    "DownBlock2D",
+    "AttnDownBlock2D",
+    "DownBlock2D"
+  ],
+  "downsample_padding": 1,
+  "downsample_type": "conv",
+  "dropout": 0.0,
+  "flip_sin_to_cos": true,
+  "freq_shift": 0,
+  "in_channels": 5,
+  "layers_per_block": 2,
+  "mid_block_scale_factor": 1,
+  "mid_block_type": "UNetMidBlock2D",
+  "norm_eps": 1e-05,
+  "norm_num_groups": 32,
+  "num_class_embeds": null,
+  "num_train_timesteps": null,
+  "out_channels": 1,
+  "resnet_time_scale_shift": "default",
+  "sample_size": 128,
+  "time_embedding_dim": null,
+  "time_embedding_type": "positional",
+  "up_block_types": [
+    "UpBlock2D",
+    "AttnUpBlock2D",
+    "UpBlock2D",
+    "UpBlock2D",
+    "UpBlock2D",
+    "UpBlock2D"
+  ],
+  "upsample_type": "conv"
+}

ddpm_model/unet/ddpm_model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4c05b76ad4d93f0897b63e513caa9acc2b888b8f414e564628466a30cde7d635
+size 454741100

fm_model/fm_model.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:16d7a540afefbb68ca98703dcb6adc574b542d4beb5f22ec2bfd25d0502b2ef0
+size 1120534858

ldm_model/ldm_model.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b33a3d77fce171a000f4c948aadaf439a18de5f4691f7d3061fdf06ec8f87d9f
+size 165808040

ldm_model/vqvae_autoencoder.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3edba23338e93457ffd2c1ce63f8631f6f25bef0ac7f172d868ec614c902be33
+size 59705428

ldm_model/vqvae_discriminator.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3476d0d59421c1d5ef28b830cc7d0a80ffec5c3154f1da26d7c2ba35d667fce4
+size 2654570

mask_model/mask_generation.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a0482056875d7646c42da7219300c8e9bd57ad5f8051b3426c0caf1adf9b85e7
+size 252760728