Upload 18 files

.gitattributes

@@ -33,3 +33,7 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+ldm_udder_v22/sample_epoch_24.png filter=lfs diff=lfs merge=lfs -text
+ldm_udder_v22/sample_epoch_49.png filter=lfs diff=lfs merge=lfs -text
+ldm_udder_v22/sample_epoch_74.png filter=lfs diff=lfs merge=lfs -text
+ldm_udder_v22/sample_epoch_99.png filter=lfs diff=lfs merge=lfs -text

ldm_udder_v22/checkpoint_epoch_24/config.json

@@ -0,0 +1,68 @@
+{
+  "_class_name": "UNet2DConditionModel",
+  "_diffusers_version": "0.36.0",
+  "_name_or_path": "ldm_udder_v21/checkpoint_epoch_99",
+  "act_fn": "silu",
+  "addition_embed_type": null,
+  "addition_embed_type_num_heads": 64,
+  "addition_time_embed_dim": null,
+  "attention_head_dim": 8,
+  "attention_type": "default",
+  "block_out_channels": [
+    128,
+    256,
+    512,
+    512
+  ],
+  "center_input_sample": false,
+  "class_embed_type": null,
+  "class_embeddings_concat": false,
+  "conv_in_kernel": 3,
+  "conv_out_kernel": 3,
+  "cross_attention_dim": 512,
+  "cross_attention_norm": null,
+  "down_block_types": [
+    "DownBlock2D",
+    "DownBlock2D",
+    "CrossAttnDownBlock2D",
+    "DownBlock2D"
+  ],
+  "downsample_padding": 1,
+  "dropout": 0.0,
+  "dual_cross_attention": false,
+  "encoder_hid_dim": null,
+  "encoder_hid_dim_type": null,
+  "flip_sin_to_cos": true,
+  "freq_shift": 0,
+  "in_channels": 4,
+  "layers_per_block": 2,
+  "mid_block_only_cross_attention": null,
+  "mid_block_scale_factor": 1,
+  "mid_block_type": "UNetMidBlock2DCrossAttn",
+  "norm_eps": 1e-05,
+  "norm_num_groups": 32,
+  "num_attention_heads": null,
+  "num_class_embeds": 5,
+  "only_cross_attention": false,
+  "out_channels": 4,
+  "projection_class_embeddings_input_dim": null,
+  "resnet_out_scale_factor": 1.0,
+  "resnet_skip_time_act": false,
+  "resnet_time_scale_shift": "default",
+  "reverse_transformer_layers_per_block": null,
+  "sample_size": 32,
+  "time_cond_proj_dim": null,
+  "time_embedding_act_fn": null,
+  "time_embedding_dim": null,
+  "time_embedding_type": "positional",
+  "timestep_post_act": null,
+  "transformer_layers_per_block": 1,
+  "up_block_types": [
+    "UpBlock2D",
+    "CrossAttnUpBlock2D",
+    "UpBlock2D",
+    "UpBlock2D"
+  ],
+  "upcast_attention": false,
+  "use_linear_projection": false
+}

ldm_udder_v22/checkpoint_epoch_24/diffusion_pytorch_model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:efe2296ffc12864edd436691c1da26b666362b576961882e2cf66897fc2cad41
+size 517986088

ldm_udder_v22/checkpoint_epoch_24/label_proj.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e22c73558eb62fb2ec130edbed659fd9fc10c65bc875b41683bae5960715b60e
+size 18924665

ldm_udder_v22/checkpoint_epoch_49/config.json

@@ -0,0 +1,68 @@
+{
+  "_class_name": "UNet2DConditionModel",
+  "_diffusers_version": "0.36.0",
+  "_name_or_path": "ldm_udder_v21/checkpoint_epoch_99",
+  "act_fn": "silu",
+  "addition_embed_type": null,
+  "addition_embed_type_num_heads": 64,
+  "addition_time_embed_dim": null,
+  "attention_head_dim": 8,
+  "attention_type": "default",
+  "block_out_channels": [
+    128,
+    256,
+    512,
+    512
+  ],
+  "center_input_sample": false,
+  "class_embed_type": null,
+  "class_embeddings_concat": false,
+  "conv_in_kernel": 3,
+  "conv_out_kernel": 3,
+  "cross_attention_dim": 512,
+  "cross_attention_norm": null,
+  "down_block_types": [
+    "DownBlock2D",
+    "DownBlock2D",
+    "CrossAttnDownBlock2D",
+    "DownBlock2D"
+  ],
+  "downsample_padding": 1,
+  "dropout": 0.0,
+  "dual_cross_attention": false,
+  "encoder_hid_dim": null,
+  "encoder_hid_dim_type": null,
+  "flip_sin_to_cos": true,
+  "freq_shift": 0,
+  "in_channels": 4,
+  "layers_per_block": 2,
+  "mid_block_only_cross_attention": null,
+  "mid_block_scale_factor": 1,
+  "mid_block_type": "UNetMidBlock2DCrossAttn",
+  "norm_eps": 1e-05,
+  "norm_num_groups": 32,
+  "num_attention_heads": null,
+  "num_class_embeds": 5,
+  "only_cross_attention": false,
+  "out_channels": 4,
+  "projection_class_embeddings_input_dim": null,
+  "resnet_out_scale_factor": 1.0,
+  "resnet_skip_time_act": false,
+  "resnet_time_scale_shift": "default",
+  "reverse_transformer_layers_per_block": null,
+  "sample_size": 32,
+  "time_cond_proj_dim": null,
+  "time_embedding_act_fn": null,
+  "time_embedding_dim": null,
+  "time_embedding_type": "positional",
+  "timestep_post_act": null,
+  "transformer_layers_per_block": 1,
+  "up_block_types": [
+    "UpBlock2D",
+    "CrossAttnUpBlock2D",
+    "UpBlock2D",
+    "UpBlock2D"
+  ],
+  "upcast_attention": false,
+  "use_linear_projection": false
+}

ldm_udder_v22/checkpoint_epoch_49/diffusion_pytorch_model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9da0a280f429fe494f4031e8498e042fb97b74314cafa0fea6108107c5d2d1a2
+size 517986088

ldm_udder_v22/checkpoint_epoch_49/label_proj.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:54fc392c5d333c4b3049681c1b892bc898527076d84f1e54e491f583435617a6
+size 18924665

ldm_udder_v22/checkpoint_epoch_74/config.json

@@ -0,0 +1,68 @@
+{
+  "_class_name": "UNet2DConditionModel",
+  "_diffusers_version": "0.36.0",
+  "_name_or_path": "ldm_udder_v21/checkpoint_epoch_99",
+  "act_fn": "silu",
+  "addition_embed_type": null,
+  "addition_embed_type_num_heads": 64,
+  "addition_time_embed_dim": null,
+  "attention_head_dim": 8,
+  "attention_type": "default",
+  "block_out_channels": [
+    128,
+    256,
+    512,
+    512
+  ],
+  "center_input_sample": false,
+  "class_embed_type": null,
+  "class_embeddings_concat": false,
+  "conv_in_kernel": 3,
+  "conv_out_kernel": 3,
+  "cross_attention_dim": 512,
+  "cross_attention_norm": null,
+  "down_block_types": [
+    "DownBlock2D",
+    "DownBlock2D",
+    "CrossAttnDownBlock2D",
+    "DownBlock2D"
+  ],
+  "downsample_padding": 1,
+  "dropout": 0.0,
+  "dual_cross_attention": false,
+  "encoder_hid_dim": null,
+  "encoder_hid_dim_type": null,
+  "flip_sin_to_cos": true,
+  "freq_shift": 0,
+  "in_channels": 4,
+  "layers_per_block": 2,
+  "mid_block_only_cross_attention": null,
+  "mid_block_scale_factor": 1,
+  "mid_block_type": "UNetMidBlock2DCrossAttn",
+  "norm_eps": 1e-05,
+  "norm_num_groups": 32,
+  "num_attention_heads": null,
+  "num_class_embeds": 5,
+  "only_cross_attention": false,
+  "out_channels": 4,
+  "projection_class_embeddings_input_dim": null,
+  "resnet_out_scale_factor": 1.0,
+  "resnet_skip_time_act": false,
+  "resnet_time_scale_shift": "default",
+  "reverse_transformer_layers_per_block": null,
+  "sample_size": 32,
+  "time_cond_proj_dim": null,
+  "time_embedding_act_fn": null,
+  "time_embedding_dim": null,
+  "time_embedding_type": "positional",
+  "timestep_post_act": null,
+  "transformer_layers_per_block": 1,
+  "up_block_types": [
+    "UpBlock2D",
+    "CrossAttnUpBlock2D",
+    "UpBlock2D",
+    "UpBlock2D"
+  ],
+  "upcast_attention": false,
+  "use_linear_projection": false
+}

ldm_udder_v22/checkpoint_epoch_74/diffusion_pytorch_model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0d12538daa50d9b835636da7320b2215b93fbaddc43c657747d6d8176715f9ef
+size 517986088

ldm_udder_v22/checkpoint_epoch_74/label_proj.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6959d36bcba9be55d52ebe61851e3573810b02e73ef9d99efd212994f5d1a273
+size 18924665

ldm_udder_v22/checkpoint_epoch_99/config.json

@@ -0,0 +1,68 @@
+{
+  "_class_name": "UNet2DConditionModel",
+  "_diffusers_version": "0.36.0",
+  "_name_or_path": "ldm_udder_v21/checkpoint_epoch_99",
+  "act_fn": "silu",
+  "addition_embed_type": null,
+  "addition_embed_type_num_heads": 64,
+  "addition_time_embed_dim": null,
+  "attention_head_dim": 8,
+  "attention_type": "default",
+  "block_out_channels": [
+    128,
+    256,
+    512,
+    512
+  ],
+  "center_input_sample": false,
+  "class_embed_type": null,
+  "class_embeddings_concat": false,
+  "conv_in_kernel": 3,
+  "conv_out_kernel": 3,
+  "cross_attention_dim": 512,
+  "cross_attention_norm": null,
+  "down_block_types": [
+    "DownBlock2D",
+    "DownBlock2D",
+    "CrossAttnDownBlock2D",
+    "DownBlock2D"
+  ],
+  "downsample_padding": 1,
+  "dropout": 0.0,
+  "dual_cross_attention": false,
+  "encoder_hid_dim": null,
+  "encoder_hid_dim_type": null,
+  "flip_sin_to_cos": true,
+  "freq_shift": 0,
+  "in_channels": 4,
+  "layers_per_block": 2,
+  "mid_block_only_cross_attention": null,
+  "mid_block_scale_factor": 1,
+  "mid_block_type": "UNetMidBlock2DCrossAttn",
+  "norm_eps": 1e-05,
+  "norm_num_groups": 32,
+  "num_attention_heads": null,
+  "num_class_embeds": 5,
+  "only_cross_attention": false,
+  "out_channels": 4,
+  "projection_class_embeddings_input_dim": null,
+  "resnet_out_scale_factor": 1.0,
+  "resnet_skip_time_act": false,
+  "resnet_time_scale_shift": "default",
+  "reverse_transformer_layers_per_block": null,
+  "sample_size": 32,
+  "time_cond_proj_dim": null,
+  "time_embedding_act_fn": null,
+  "time_embedding_dim": null,
+  "time_embedding_type": "positional",
+  "timestep_post_act": null,
+  "transformer_layers_per_block": 1,
+  "up_block_types": [
+    "UpBlock2D",
+    "CrossAttnUpBlock2D",
+    "UpBlock2D",
+    "UpBlock2D"
+  ],
+  "upcast_attention": false,
+  "use_linear_projection": false
+}

ldm_udder_v22/checkpoint_epoch_99/diffusion_pytorch_model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:28f5a2d0a89149235228fc0f4c6da1efabe0a06985559e6e433eddbbe18538e0
+size 517986088

ldm_udder_v22/checkpoint_epoch_99/label_proj.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9ff6d5eff1f602519f5ca6f03b59fee1bfb47d286e96bf402bdb5fc76805e6a7
+size 18924665

ldm_udder_v22/train_log.txt

@@ -0,0 +1,100 @@
+Epoch 0: Avg Loss = 0.080715
+Epoch 1: Avg Loss = 0.074514
+Epoch 2: Avg Loss = 0.075358
+Epoch 3: Avg Loss = 0.075383
+Epoch 4: Avg Loss = 0.076274
+Epoch 5: Avg Loss = 0.075728
+Epoch 6: Avg Loss = 0.076468
+Epoch 7: Avg Loss = 0.082177
+Epoch 8: Avg Loss = 0.075511
+Epoch 9: Avg Loss = 0.079308
+Epoch 10: Avg Loss = 0.075315
+Epoch 11: Avg Loss = 0.084014
+Epoch 12: Avg Loss = 0.082420
+Epoch 13: Avg Loss = 0.085380
+Epoch 14: Avg Loss = 0.085959
+Epoch 15: Avg Loss = 0.086952
+Epoch 16: Avg Loss = 0.081588
+Epoch 17: Avg Loss = 0.082715
+Epoch 18: Avg Loss = 0.092930
+Epoch 19: Avg Loss = 0.083338
+Epoch 20: Avg Loss = 0.086847
+Epoch 21: Avg Loss = 0.090454
+Epoch 22: Avg Loss = 0.088372
+Epoch 23: Avg Loss = 0.090116
+Epoch 24: Avg Loss = 0.102798
+Epoch 25: Avg Loss = 0.088631
+Epoch 26: Avg Loss = 0.099135
+Epoch 27: Avg Loss = 0.087258
+Epoch 28: Avg Loss = 0.097918
+Epoch 29: Avg Loss = 0.089590
+Epoch 30: Avg Loss = 0.094060
+Epoch 31: Avg Loss = 0.095418
+Epoch 32: Avg Loss = 0.094168
+Epoch 33: Avg Loss = 0.102046
+Epoch 34: Avg Loss = 0.101552
+Epoch 35: Avg Loss = 0.099833
+Epoch 36: Avg Loss = 0.102155
+Epoch 37: Avg Loss = 0.097737
+Epoch 38: Avg Loss = 0.109230
+Epoch 39: Avg Loss = 0.103714
+Epoch 40: Avg Loss = 0.100241
+Epoch 41: Avg Loss = 0.106330
+Epoch 42: Avg Loss = 0.104612
+Epoch 43: Avg Loss = 0.108008
+Epoch 44: Avg Loss = 0.108921
+Epoch 45: Avg Loss = 0.105485
+Epoch 46: Avg Loss = 0.101489
+Epoch 47: Avg Loss = 0.115491
+Epoch 48: Avg Loss = 0.113348
+Epoch 49: Avg Loss = 0.113600
+Epoch 50: Avg Loss = 0.116021
+Epoch 51: Avg Loss = 0.107912
+Epoch 52: Avg Loss = 0.114556
+Epoch 53: Avg Loss = 0.113709
+Epoch 54: Avg Loss = 0.114417
+Epoch 55: Avg Loss = 0.118580
+Epoch 56: Avg Loss = 0.117411
+Epoch 57: Avg Loss = 0.122868
+Epoch 58: Avg Loss = 0.121349
+Epoch 59: Avg Loss = 0.114866
+Epoch 60: Avg Loss = 0.119930
+Epoch 61: Avg Loss = 0.114256
+Epoch 62: Avg Loss = 0.118595
+Epoch 63: Avg Loss = 0.126395
+Epoch 64: Avg Loss = 0.123684
+Epoch 65: Avg Loss = 0.124535
+Epoch 66: Avg Loss = 0.119938
+Epoch 67: Avg Loss = 0.130023
+Epoch 68: Avg Loss = 0.126239
+Epoch 69: Avg Loss = 0.127046
+Epoch 70: Avg Loss = 0.123564
+Epoch 71: Avg Loss = 0.129876
+Epoch 72: Avg Loss = 0.135812
+Epoch 73: Avg Loss = 0.131150
+Epoch 74: Avg Loss = 0.129281
+Epoch 75: Avg Loss = 0.129877
+Epoch 76: Avg Loss = 0.134494
+Epoch 77: Avg Loss = 0.136164
+Epoch 78: Avg Loss = 0.131985
+Epoch 79: Avg Loss = 0.133029
+Epoch 80: Avg Loss = 0.137856
+Epoch 81: Avg Loss = 0.138163
+Epoch 82: Avg Loss = 0.138714
+Epoch 83: Avg Loss = 0.139421
+Epoch 84: Avg Loss = 0.137069
+Epoch 85: Avg Loss = 0.140045
+Epoch 86: Avg Loss = 0.139484
+Epoch 87: Avg Loss = 0.148398
+Epoch 88: Avg Loss = 0.144912
+Epoch 89: Avg Loss = 0.148541
+Epoch 90: Avg Loss = 0.147067
+Epoch 91: Avg Loss = 0.144395
+Epoch 92: Avg Loss = 0.146554
+Epoch 93: Avg Loss = 0.142251
+Epoch 94: Avg Loss = 0.149697
+Epoch 95: Avg Loss = 0.153768
+Epoch 96: Avg Loss = 0.143709
+Epoch 97: Avg Loss = 0.147047
+Epoch 98: Avg Loss = 0.154985
+Epoch 99: Avg Loss = 0.145946