Initial commit

README.md

@@ -1,3 +1,5 @@
 ---
 license: cc-by-nc-4.0
+tags:
+    - unconditional-image-generation
 ---

checkpoint-180000/optimizer.bin

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

checkpoint-180000/random_states_0.pkl

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

checkpoint-180000/scheduler.bin

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

checkpoint-180000/unet/config.json

@@ -0,0 +1,46 @@
+{
+  "_class_name": "UNet2DModel",
+  "_diffusers_version": "0.15.1",
+  "act_fn": "silu",
+  "add_attention": true,
+  "attention_head_dim": 8,
+  "block_out_channels": [
+    32,
+    64,
+    64,
+    64,
+    128,
+    256
+  ],
+  "center_input_sample": false,
+  "class_embed_type": null,
+  "down_block_types": [
+    "DownBlock2D",
+    "DownBlock2D",
+    "DownBlock2D",
+    "DownBlock2D",
+    "AttnDownBlock2D",
+    "DownBlock2D"
+  ],
+  "downsample_padding": 1,
+  "flip_sin_to_cos": true,
+  "freq_shift": 0,
+  "in_channels": 1,
+  "layers_per_block": 1,
+  "mid_block_scale_factor": 1,
+  "norm_eps": 1e-05,
+  "norm_num_groups": 32,
+  "num_class_embeds": null,
+  "out_channels": 1,
+  "resnet_time_scale_shift": "default",
+  "sample_size": 512,
+  "time_embedding_type": "positional",
+  "up_block_types": [
+    "UpBlock2D",
+    "AttnUpBlock2D",
+    "UpBlock2D",
+    "UpBlock2D",
+    "UpBlock2D",
+    "UpBlock2D"
+  ]
+}

checkpoint-180000/unet/diffusion_pytorch_model.bin

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

checkpoint-180000/unet_ema/config.json

@@ -0,0 +1,53 @@
+{
+  "_class_name": "UNet2DModel",
+  "_diffusers_version": "0.15.1",
+  "act_fn": "silu",
+  "add_attention": true,
+  "attention_head_dim": 8,
+  "block_out_channels": [
+    32,
+    64,
+    64,
+    64,
+    128,
+    256
+  ],
+  "center_input_sample": false,
+  "class_embed_type": null,
+  "decay": 0.9999,
+  "down_block_types": [
+    "DownBlock2D",
+    "DownBlock2D",
+    "DownBlock2D",
+    "DownBlock2D",
+    "AttnDownBlock2D",
+    "DownBlock2D"
+  ],
+  "downsample_padding": 1,
+  "flip_sin_to_cos": true,
+  "freq_shift": 0,
+  "in_channels": 1,
+  "inv_gamma": 1.0,
+  "layers_per_block": 1,
+  "mid_block_scale_factor": 1,
+  "min_decay": 0.0,
+  "norm_eps": 1e-05,
+  "norm_num_groups": 32,
+  "num_class_embeds": null,
+  "optimization_step": 180000,
+  "out_channels": 1,
+  "power": 0.75,
+  "resnet_time_scale_shift": "default",
+  "sample_size": 512,
+  "time_embedding_type": "positional",
+  "up_block_types": [
+    "UpBlock2D",
+    "AttnUpBlock2D",
+    "UpBlock2D",
+    "UpBlock2D",
+    "UpBlock2D",
+    "UpBlock2D"
+  ],
+  "update_after_step": 0,
+  "use_ema_warmup": true
+}

checkpoint-180000/unet_ema/diffusion_pytorch_model.bin

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

model_index.json

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

scheduler/scheduler_config.json

@@ -0,0 +1,16 @@
+{
+  "_class_name": "DDPMScheduler",
+  "_diffusers_version": "0.15.1",
+  "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",
+  "sample_max_value": 1.0,
+  "thresholding": false,
+  "trained_betas": null,
+  "variance_type": "fixed_small"
+}

unet/config.json

@@ -0,0 +1,46 @@
+{
+  "_class_name": "UNet2DModel",
+  "_diffusers_version": "0.15.1",
+  "act_fn": "silu",
+  "add_attention": true,
+  "attention_head_dim": 8,
+  "block_out_channels": [
+    32,
+    64,
+    64,
+    64,
+    128,
+    256
+  ],
+  "center_input_sample": false,
+  "class_embed_type": null,
+  "down_block_types": [
+    "DownBlock2D",
+    "DownBlock2D",
+    "DownBlock2D",
+    "DownBlock2D",
+    "AttnDownBlock2D",
+    "DownBlock2D"
+  ],
+  "downsample_padding": 1,
+  "flip_sin_to_cos": true,
+  "freq_shift": 0,
+  "in_channels": 1,
+  "layers_per_block": 1,
+  "mid_block_scale_factor": 1,
+  "norm_eps": 1e-05,
+  "norm_num_groups": 32,
+  "num_class_embeds": null,
+  "out_channels": 1,
+  "resnet_time_scale_shift": "default",
+  "sample_size": 512,
+  "time_embedding_type": "positional",
+  "up_block_types": [
+    "UpBlock2D",
+    "AttnUpBlock2D",
+    "UpBlock2D",
+    "UpBlock2D",
+    "UpBlock2D",
+    "UpBlock2D"
+  ]
+}

unet/diffusion_pytorch_model.bin

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