sebascorreia commited on
Commit
53e5548
·
verified ·
1 Parent(s): 87dce1d
model_index.json ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "DDPMPipeline",
3
+ "_diffusers_version": "0.24.0",
4
+ "scheduler": [
5
+ "diffusers",
6
+ "DDPMScheduler"
7
+ ],
8
+ "unet": [
9
+ "diffusers",
10
+ "UNet2DModel"
11
+ ]
12
+ }
scheduler/scheduler_config.json ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "DDPMScheduler",
3
+ "_diffusers_version": "0.24.0",
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
+ "steps_offset": 0,
14
+ "thresholding": false,
15
+ "timestep_spacing": "leading",
16
+ "trained_betas": null,
17
+ "variance_type": "fixed_small"
18
+ }
tensorboard/train_example/events.out.tfevents.1725325228.62e6b81354d1.143564.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:93e708b26a217dcf52595c9cbf3254056a2273ec64ee8b7287049d396315b34a
3
+ size 88
tensorboard/train_example/events.out.tfevents.1725325246.62e6b81354d1.143705.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8092af057f812799f601276c22a61b7e827b77b61b9b44a2985829322606b806
3
+ size 87261
unet/config.json ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DModel",
3
+ "_diffusers_version": "0.24.0",
4
+ "_name_or_path": "/root/.cache/huggingface/hub/models--sebascorreia--DDPM-sc09-conditional/snapshots/cd30cddc1244e17ceadbf92814a2dc7702b99b2d/unet",
5
+ "act_fn": "silu",
6
+ "add_attention": true,
7
+ "attention_head_dim": 8,
8
+ "attn_norm_num_groups": null,
9
+ "block_out_channels": [
10
+ 128,
11
+ 128,
12
+ 256,
13
+ 256,
14
+ 512,
15
+ 512
16
+ ],
17
+ "center_input_sample": false,
18
+ "class_embed_type": null,
19
+ "down_block_types": [
20
+ "DownBlock2D",
21
+ "DownBlock2D",
22
+ "DownBlock2D",
23
+ "DownBlock2D",
24
+ "AttnDownBlock2D",
25
+ "DownBlock2D"
26
+ ],
27
+ "downsample_padding": 1,
28
+ "downsample_type": "conv",
29
+ "dropout": 0.0,
30
+ "flip_sin_to_cos": true,
31
+ "freq_shift": 0,
32
+ "in_channels": 1,
33
+ "layers_per_block": 2,
34
+ "mid_block_scale_factor": 1,
35
+ "norm_eps": 1e-05,
36
+ "norm_num_groups": 32,
37
+ "num_class_embeds": 10,
38
+ "num_train_timesteps": null,
39
+ "out_channels": 1,
40
+ "resnet_time_scale_shift": "default",
41
+ "sample_size": 128,
42
+ "time_embedding_type": "positional",
43
+ "up_block_types": [
44
+ "UpBlock2D",
45
+ "AttnUpBlock2D",
46
+ "UpBlock2D",
47
+ "UpBlock2D",
48
+ "UpBlock2D",
49
+ "UpBlock2D"
50
+ ],
51
+ "upsample_type": "conv"
52
+ }
unet/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4513d0e24960aabb9a92a8571f54401b32ab59a11eeff0be9d5ef995a8aab264
3
+ size 454743236