zivzhong commited on
Commit
dbca116
1 Parent(s): 1185751
README.md ADDED
@@ -0,0 +1,54 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language: en
3
+ license: apache-2.0
4
+ library_name: diffusers
5
+ tags: []
6
+ datasets: huggan/smithsonian_butterflies_subset
7
+ metrics: []
8
+ ---
9
+
10
+ <!-- This model card has been generated automatically according to the information the training script had access to. You
11
+ should probably proofread and complete it, then remove this comment. -->
12
+
13
+ # ddpm-butterflies-128
14
+
15
+ ## Model description
16
+
17
+ This diffusion model is trained with the [🤗 Diffusers](https://github.com/huggingface/diffusers) library
18
+ on the `huggan/smithsonian_butterflies_subset` dataset.
19
+
20
+ ## Intended uses & limitations
21
+
22
+ #### How to use
23
+
24
+ ```python
25
+ # TODO: add an example code snippet for running this diffusion pipeline
26
+ ```
27
+
28
+ #### Limitations and bias
29
+
30
+ [TODO: provide examples of latent issues and potential remediations]
31
+
32
+ ## Training data
33
+
34
+ [TODO: describe the data used to train the model]
35
+
36
+ ### Training hyperparameters
37
+
38
+ The following hyperparameters were used during training:
39
+ - learning_rate: 0.0001
40
+ - train_batch_size: 16
41
+ - eval_batch_size: 16
42
+ - gradient_accumulation_steps: 1
43
+ - optimizer: AdamW with betas=(None, None), weight_decay=None and epsilon=None
44
+ - lr_scheduler: None
45
+ - lr_warmup_steps: 500
46
+ - ema_inv_gamma: None
47
+ - ema_inv_gamma: None
48
+ - ema_inv_gamma: None
49
+ - mixed_precision: fp16
50
+
51
+ ### Training results
52
+
53
+ 📈 [TensorBoard logs](https://huggingface.co/HuggingFace7/ddpm-butterflies-128/tensorboard?#scalars)
54
+
logs/train_example/events.out.tfevents.1671073010.a3cdbf167961.72.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3c28a2041055513e7f0485e5a9074d803b1db75bd6562adfea8ea9908df392d6
3
+ size 9706
logs/train_example/events.out.tfevents.1671073148.a3cdbf167961.72.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:14f4bed2379814dba8fb6e610279fb97f76e1a503f1778926cf37ea950fdd25f
3
+ size 79660
logs/train_example/events.out.tfevents.1671084391.c4da25ab7987.74.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5105bc0996a84e6d82e3836ae2015847ce720941ac249f98f605c9c3338f9724
3
+ size 23653
logs/train_example/events.out.tfevents.1706769159.10d90e8dc3e6.217.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:05f0354f96ebbcdcad0738deaec5534c9024a3137401062c62de473c9270ee94
3
+ size 239728
model_index.json ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "DDPMPipeline",
3
+ "_diffusers_version": "0.5.1",
4
+ "scheduler": [
5
+ "diffusers",
6
+ "DDPMScheduler"
7
+ ],
8
+ "unet": [
9
+ "diffusers",
10
+ "UNet2DModel"
11
+ ]
12
+ }
samples/0002.png ADDED
samples/0009.png ADDED
samples/0019.png ADDED
samples/0029.png ADDED
scheduler/scheduler_config.json ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "DDPMScheduler",
3
+ "_diffusers_version": "0.5.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_small"
11
+ }
unet/config.json ADDED
@@ -0,0 +1,42 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DModel",
3
+ "_diffusers_version": "0.5.1",
4
+ "act_fn": "silu",
5
+ "attention_head_dim": 8,
6
+ "block_out_channels": [
7
+ 128,
8
+ 128,
9
+ 256,
10
+ 256,
11
+ 512,
12
+ 512
13
+ ],
14
+ "center_input_sample": false,
15
+ "down_block_types": [
16
+ "DownBlock2D",
17
+ "DownBlock2D",
18
+ "DownBlock2D",
19
+ "DownBlock2D",
20
+ "AttnDownBlock2D",
21
+ "DownBlock2D"
22
+ ],
23
+ "downsample_padding": 1,
24
+ "flip_sin_to_cos": true,
25
+ "freq_shift": 0,
26
+ "in_channels": 3,
27
+ "layers_per_block": 2,
28
+ "mid_block_scale_factor": 1,
29
+ "norm_eps": 1e-05,
30
+ "norm_num_groups": 32,
31
+ "out_channels": 3,
32
+ "sample_size": 128,
33
+ "time_embedding_type": "positional",
34
+ "up_block_types": [
35
+ "UpBlock2D",
36
+ "AttnUpBlock2D",
37
+ "UpBlock2D",
38
+ "UpBlock2D",
39
+ "UpBlock2D",
40
+ "UpBlock2D"
41
+ ]
42
+ }
unet/diffusion_pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:428e41bf9b58bd80230e8dd93f1ba40377ce54bb3eb9e31c6e482aa59d651217
3
+ size 454867325