Hugging Face's logo

giulio98
/
celeba_processed_conditional_hair_color

Diffusers
Safetensors
ScoreSdeVePipelineConditioned
Model card Files Community
giulio98 commited on
Commit
8012106
•
1 Parent(s): 0bc882e

Epoch 511

Browse files
Files changed (9) hide show
  1. checkpoint-325540/optimizer.bin +3 -0
  2. checkpoint-325540/random_states_0.pkl +3 -0
  3. checkpoint-325540/scaler.pt +3 -0
  4. checkpoint-325540/scheduler.bin +3 -0
  5. checkpoint-325540/unet/config.json +55 -0
  6. checkpoint-325540/unet/diffusion_pytorch_model.safetensors +3 -0
  7. checkpoint-325540/unet_ema/config.json +62 -0
  8. checkpoint-325540/unet_ema/diffusion_pytorch_model.safetensors +3 -0
  9. unet/diffusion_pytorch_model.safetensors +1 -1
checkpoint-325540/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1f61ba36df8ee2079fdf0160cc2edd49e96b33a9006510bbfde41a6efc149fc6
3
+ size 525162465
checkpoint-325540/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:22f75caff47a6a56d29750a01a339e74092375b5bff361616853e96d951a665d
3
+ size 14344
checkpoint-325540/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:86bee277c47d0d2546664aa2e1763fb58c8e799c08c1171b11126c1242189cdf
3
+ size 988
checkpoint-325540/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:28baaeb02f4f53735472b3ed8a0265a20cb46476f41865178162bc807309a1a5
3
+ size 1000
checkpoint-325540/unet/config.json ADDED
@@ -0,0 +1,55 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DModel",
3
+ "_diffusers_version": "0.27.2",
4
+ "act_fn": "silu",
5
+ "add_attention": true,
6
+ "attention_head_dim": null,
7
+ "attn_norm_num_groups": null,
8
+ "block_out_channels": [
9
+ 128,
10
+ 128,
11
+ 256,
12
+ 256,
13
+ 256,
14
+ 256,
15
+ 256
16
+ ],
17
+ "center_input_sample": true,
18
+ "class_embed_type": null,
19
+ "down_block_types": [
20
+ "SkipDownBlock2D",
21
+ "SkipDownBlock2D",
22
+ "SkipDownBlock2D",
23
+ "SkipDownBlock2D",
24
+ "AttnSkipDownBlock2D",
25
+ "SkipDownBlock2D",
26
+ "SkipDownBlock2D"
27
+ ],
28
+ "downsample_padding": 1,
29
+ "downsample_type": "conv",
30
+ "dropout": 0.0,
31
+ "flip_sin_to_cos": true,
32
+ "freq_shift": 0,
33
+ "in_channels": 3,
34
+ "layers_per_block": 2,
35
+ "mid_block_scale_factor": 1.41421356237,
36
+ "norm_eps": 1e-06,
37
+ "norm_num_groups": null,
38
+ "num_class_embeds": 6,
39
+ "num_train_timesteps": null,
40
+ "out_channels": 3,
41
+ "resnet_time_scale_shift": "default",
42
+ "sample_size": 64,
43
+ "set_W_to_weight": false,
44
+ "time_embedding_type": "fourier",
45
+ "up_block_types": [
46
+ "SkipUpBlock2D",
47
+ "SkipUpBlock2D",
48
+ "AttnSkipUpBlock2D",
49
+ "SkipUpBlock2D",
50
+ "SkipUpBlock2D",
51
+ "SkipUpBlock2D",
52
+ "SkipUpBlock2D"
53
+ ],
54
+ "upsample_type": "conv"
55
+ }
checkpoint-325540/unet/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e6998f97866dc8c355969e8a52f2cc7fb22c2989562f3f886c233deb695317cb
3
+ size 262379380
checkpoint-325540/unet_ema/config.json ADDED
@@ -0,0 +1,62 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DModel",
3
+ "_diffusers_version": "0.27.2",
4
+ "act_fn": "silu",
5
+ "add_attention": true,
6
+ "attention_head_dim": null,
7
+ "attn_norm_num_groups": null,
8
+ "block_out_channels": [
9
+ 128,
10
+ 128,
11
+ 256,
12
+ 256,
13
+ 256,
14
+ 256,
15
+ 256
16
+ ],
17
+ "center_input_sample": true,
18
+ "class_embed_type": null,
19
+ "decay": 0.9999,
20
+ "down_block_types": [
21
+ "SkipDownBlock2D",
22
+ "SkipDownBlock2D",
23
+ "SkipDownBlock2D",
24
+ "SkipDownBlock2D",
25
+ "AttnSkipDownBlock2D",
26
+ "SkipDownBlock2D",
27
+ "SkipDownBlock2D"
28
+ ],
29
+ "downsample_padding": 1,
30
+ "downsample_type": "conv",
31
+ "dropout": 0.0,
32
+ "flip_sin_to_cos": true,
33
+ "freq_shift": 0,
34
+ "in_channels": 3,
35
+ "inv_gamma": 1.0,
36
+ "layers_per_block": 2,
37
+ "mid_block_scale_factor": 1.41421356237,
38
+ "min_decay": 0.0,
39
+ "norm_eps": 1e-06,
40
+ "norm_num_groups": null,
41
+ "num_class_embeds": 6,
42
+ "num_train_timesteps": null,
43
+ "optimization_step": 325540,
44
+ "out_channels": 3,
45
+ "power": 0.75,
46
+ "resnet_time_scale_shift": "default",
47
+ "sample_size": 64,
48
+ "set_W_to_weight": false,
49
+ "time_embedding_type": "fourier",
50
+ "up_block_types": [
51
+ "SkipUpBlock2D",
52
+ "SkipUpBlock2D",
53
+ "AttnSkipUpBlock2D",
54
+ "SkipUpBlock2D",
55
+ "SkipUpBlock2D",
56
+ "SkipUpBlock2D",
57
+ "SkipUpBlock2D"
58
+ ],
59
+ "update_after_step": 0,
60
+ "upsample_type": "conv",
61
+ "use_ema_warmup": true
62
+ }
checkpoint-325540/unet_ema/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:65f19f308e41d4411b7e017ea849c8d0486f3e71d025e964706020f15bc8d94b
3
+ size 262379380
unet/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6be7f5c26f6dff717ea3cd8513f598a5039ac0034930f911152f8aa8f663a885
3
  size 262379380
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ee7552089b838db47d177c25feb8c6946605902c137fd21e378310fb50714f03
3
  size 262379380