Artiprocher commited on
Commit
1930f29
1 Parent(s): 86c7d83

update model

Browse files
model.ckpt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e66e873771ab03b76b9347983acaa265f3e84e7b385eefd3992a0b4a61ebfe8f
3
  size 4178303774
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d1cb551139fbbdb7cbb2c8f0092e10909fb00e7e923400c532d223b03bd4ca44
3
  size 4178303774
unet/config.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "_class_name": "UNet2DConditionModel",
3
  "_diffusers_version": "0.15.0.dev0",
4
- "_name_or_path": "models/20230303_food/checkpoint-35000/unet",
5
  "act_fn": "silu",
6
  "attention_head_dim": 8,
7
  "block_out_channels": [
@@ -15,6 +15,7 @@
15
  "conv_in_kernel": 3,
16
  "conv_out_kernel": 3,
17
  "cross_attention_dim": 768,
 
18
  "down_block_types": [
19
  "CrossAttnDownBlock2D",
20
  "CrossAttnDownBlock2D",
@@ -26,14 +27,18 @@
26
  "flip_sin_to_cos": true,
27
  "freq_shift": 0,
28
  "in_channels": 4,
 
29
  "layers_per_block": 2,
30
  "mid_block_scale_factor": 1,
31
  "mid_block_type": "UNetMidBlock2DCrossAttn",
 
32
  "norm_eps": 1e-05,
33
  "norm_num_groups": 32,
34
  "num_class_embeds": null,
35
  "only_cross_attention": false,
 
36
  "out_channels": 4,
 
37
  "projection_class_embeddings_input_dim": null,
38
  "resnet_time_scale_shift": "default",
39
  "sample_size": 64,
@@ -47,5 +52,7 @@
47
  "CrossAttnUpBlock2D"
48
  ],
49
  "upcast_attention": false,
 
 
50
  "use_linear_projection": false
51
  }
 
1
  {
2
  "_class_name": "UNet2DConditionModel",
3
  "_diffusers_version": "0.15.0.dev0",
4
+ "_name_or_path": "models/20230306_food/checkpoint-95000/unet_ema",
5
  "act_fn": "silu",
6
  "attention_head_dim": 8,
7
  "block_out_channels": [
 
15
  "conv_in_kernel": 3,
16
  "conv_out_kernel": 3,
17
  "cross_attention_dim": 768,
18
+ "decay": 0.9999,
19
  "down_block_types": [
20
  "CrossAttnDownBlock2D",
21
  "CrossAttnDownBlock2D",
 
27
  "flip_sin_to_cos": true,
28
  "freq_shift": 0,
29
  "in_channels": 4,
30
+ "inv_gamma": 1.0,
31
  "layers_per_block": 2,
32
  "mid_block_scale_factor": 1,
33
  "mid_block_type": "UNetMidBlock2DCrossAttn",
34
+ "min_decay": 0.0,
35
  "norm_eps": 1e-05,
36
  "norm_num_groups": 32,
37
  "num_class_embeds": null,
38
  "only_cross_attention": false,
39
+ "optimization_step": 95000,
40
  "out_channels": 4,
41
+ "power": 0.6666666666666666,
42
  "projection_class_embeddings_input_dim": null,
43
  "resnet_time_scale_shift": "default",
44
  "sample_size": 64,
 
52
  "CrossAttnUpBlock2D"
53
  ],
54
  "upcast_attention": false,
55
+ "update_after_step": 0,
56
+ "use_ema_warmup": false,
57
  "use_linear_projection": false
58
  }
unet/diffusion_pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:bc7325ad594d2f48a26cab4d93167672c6455a1581e9eabab0033fe74cdfe205
3
  size 3438354725
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:76ab5f573ccf25e9c76e65a2d88843526a9ada929f9dd7a9102629231c6124a5
3
  size 3438354725