End of training

lr_scheduler/lr_scheduler.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9bb41cb3d42a6d56256790e694f437ac0e1479daf9a416e5bc0053209e0a4d46
+oid sha256:9d20c2063d5e60005b2a44a7119f69299af9886b73adf1efae8603ea11217c3f
 size 1076

optimizer/optimizer.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:11e9c2f4a33917b68698782dbd28529af6674bf8419c8306805571af782700a9
-size 1149290750
+oid sha256:d5b732b08f51120e889a70491395c816c537e0d02aa7b632977164dcb3e5fa8a
+size 6876705882

scheduler/sdeparameters.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8d080a1e759b62949da6079135e99c5d2d09b8a682635e7d78fab71fc16a69d5
+oid sha256:06d4e02865d20fa3ab8b76066eb2dc3b5b2b260175b485a3421b2de143e88e45
 size 136

unet/config.json

@@ -8,12 +8,10 @@
   "attention_head_dim": 8,
   "attention_type": "default",
   "block_out_channels": [
-    128,
-    128,
-    256,
-    256,
-    512,
-    512
+    320,
+    640,
+    1280,
+    1280
   ],
   "center_input_sample": false,
   "class_embed_type": null,
@@ -23,10 +21,8 @@
   "cross_attention_dim": 768,
   "cross_attention_norm": null,
   "down_block_types": [
-    "DownBlock2D",
-    "DownBlock2D",
-    "DownBlock2D",
-    "DownBlock2D",
+    "CrossAttnDownBlock2D",
+    "CrossAttnDownBlock2D",
     "CrossAttnDownBlock2D",
     "DownBlock2D"
   ],
@@ -63,10 +59,8 @@
   "up_block_types": [
     "UpBlock2D",
     "CrossAttnUpBlock2D",
-    "UpBlock2D",
-    "UpBlock2D",
-    "UpBlock2D",
-    "UpBlock2D"
+    "CrossAttnUpBlock2D",
+    "CrossAttnUpBlock2D"
   ],
   "upcast_attention": false,
   "use_linear_projection": false

unet/diffusion_pytorch_model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:2c1c80505b7fc5fdc7c447e52460f08b5f88e4e61f971b8e4391f615eef96d2f
-size 574476604
+oid sha256:dce9d4e63f7a84dc3aa5232f4aa14582a59cb1c6eb81d610b84f481820a69d36
+size 3438144492