patrickvonplaten commited on
Commit
e3add20
1 Parent(s): 96eb466
model_index.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "_class_name": "DDPMPipeline",
3
- "_module": "modeling_ddpm.py",
4
  "scheduler": [
5
  "diffusers",
6
  "DDPMScheduler"
1
  {
2
  "_class_name": "DDPMPipeline",
3
+ "_diffusers_version": "0.15.0.dev0",
4
  "scheduler": [
5
  "diffusers",
6
  "DDPMScheduler"
scheduler_config.json → scheduler/scheduler_config.json RENAMED
@@ -1,11 +1,16 @@
1
  {
2
  "_class_name": "DDPMScheduler",
3
- "_diffusers_version": "0.1.1",
4
  "beta_end": 0.02,
5
  "beta_schedule": "linear",
6
  "beta_start": 0.0001,
7
  "clip_sample": true,
 
 
8
  "num_train_timesteps": 1000,
 
 
 
9
  "trained_betas": null,
10
  "variance_type": "fixed_large"
11
  }
1
  {
2
  "_class_name": "DDPMScheduler",
3
+ "_diffusers_version": "0.15.0.dev0",
4
  "beta_end": 0.02,
5
  "beta_schedule": "linear",
6
  "beta_start": 0.0001,
7
  "clip_sample": true,
8
+ "clip_sample_range": 1.0,
9
+ "dynamic_thresholding_ratio": 0.995,
10
  "num_train_timesteps": 1000,
11
+ "prediction_type": "epsilon",
12
+ "sample_max_value": 1.0,
13
+ "thresholding": false,
14
  "trained_betas": null,
15
  "variance_type": "fixed_large"
16
  }
config.json → unet/config.json RENAMED
@@ -1,7 +1,9 @@
1
  {
2
  "_class_name": "UNet2DModel",
3
- "_diffusers_version": "0.0.4",
 
4
  "act_fn": "silu",
 
5
  "attention_head_dim": null,
6
  "block_out_channels": [
7
  128,
@@ -10,6 +12,7 @@
10
  256
11
  ],
12
  "center_input_sample": false,
 
13
  "down_block_types": [
14
  "DownBlock2D",
15
  "AttnDownBlock2D",
@@ -24,7 +27,9 @@
24
  "mid_block_scale_factor": 1,
25
  "norm_eps": 1e-06,
26
  "norm_num_groups": 32,
 
27
  "out_channels": 3,
 
28
  "sample_size": 32,
29
  "time_embedding_type": "positional",
30
  "up_block_types": [
1
  {
2
  "_class_name": "UNet2DModel",
3
+ "_diffusers_version": "0.15.0.dev0",
4
+ "_name_or_path": "./",
5
  "act_fn": "silu",
6
+ "add_attention": true,
7
  "attention_head_dim": null,
8
  "block_out_channels": [
9
  128,
12
  256
13
  ],
14
  "center_input_sample": false,
15
+ "class_embed_type": null,
16
  "down_block_types": [
17
  "DownBlock2D",
18
  "AttnDownBlock2D",
27
  "mid_block_scale_factor": 1,
28
  "norm_eps": 1e-06,
29
  "norm_num_groups": 32,
30
+ "num_class_embeds": null,
31
  "out_channels": 3,
32
+ "resnet_time_scale_shift": "default",
33
  "sample_size": 32,
34
  "time_embedding_type": "positional",
35
  "up_block_types": [
diffusion_pytorch_model.bin → unet/diffusion_pytorch_model.bin RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ac3416a548879738893e935b42cab9119f51745b62bec4b6d8375e4d86e98ea4
3
- size 143101489
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:05f238efa9d7877867ae2d75efedf50044de988d7f5d827e12cd1b51c5c51fda
3
+ size 143107409