Epoch 1

logs/train_example/events.out.tfevents.1721839137.619684b2f747.34.2

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b29de647e2ca984ee76c11151646f1ad1082a9214adb1b2c22bd338c6eb67cff
-size 15582
+oid sha256:f3a878d351406392a8b401fd7c03adffc07cac20b3d8e7fb9955e7cd71afd661
+size 31448

unet/config.json

@@ -6,11 +6,12 @@
   "attention_head_dim": 8,
   "attn_norm_num_groups": null,
   "block_out_channels": [
-    32,
-    64,
-    64,
     128,
-    256
+    128,
+    256,
+    256,
+    512,
+    512
   ],
   "center_input_sample": false,
   "class_embed_type": null,
@@ -19,7 +20,8 @@
     "DownBlock2D",
     "DownBlock2D",
     "DownBlock2D",
-    "AttnDownBlock2D"
+    "AttnDownBlock2D",
+    "DownBlock2D"
   ],
   "downsample_padding": 1,
   "downsample_type": "conv",
@@ -41,6 +43,7 @@
   ],
   "time_embedding_type": "positional",
   "up_block_types": [
+    "UpBlock2D",
     "AttnUpBlock2D",
     "UpBlock2D",
     "UpBlock2D",

unet/diffusion_pytorch_model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:bc994fe7cd4242fae4192a04c0f7006b510305a8a02a7d6e9ac87987aebc9019
-size 64432932
+oid sha256:400902fe54a25a3220704c72f86aef297dc8f42644f82ec56d4ae14ef0f88b09
+size 454741108