End of training

model_index.json

@@ -1,6 +1,6 @@
 {
   "_class_name": "UTTIPipeline",
-  "_diffusers_version": "0.24.0.dev0",
+  "_diffusers_version": "0.25.0.dev0",
   "scheduler": [
     null,
     null
@@ -10,8 +10,8 @@
     "CLIPTextModel"
   ],
   "tokenizer": [
-    null,
-    null
+    "transformers",
+    "CLIPTokenizer"
   ],
   "unet": [
     "diffusers",

unet/config.json

@@ -1,6 +1,6 @@
 {
   "_class_name": "UNet2DConditionModel",
-  "_diffusers_version": "0.24.0.dev0",
+  "_diffusers_version": "0.25.0.dev0",
   "act_fn": "silu",
   "addition_embed_type": null,
   "addition_embed_type_num_heads": 64,
@@ -8,10 +8,12 @@
   "attention_head_dim": 8,
   "attention_type": "default",
   "block_out_channels": [
-    320,
-    640,
-    1280,
-    1280
+    128,
+    128,
+    256,
+    256,
+    512,
+    512
   ],
   "center_input_sample": false,
   "class_embed_type": null,
@@ -21,8 +23,10 @@
   "cross_attention_dim": 768,
   "cross_attention_norm": null,
   "down_block_types": [
-    "CrossAttnDownBlock2D",
-    "CrossAttnDownBlock2D",
+    "DownBlock2D",
+    "DownBlock2D",
+    "DownBlock2D",
+    "DownBlock2D",
     "CrossAttnDownBlock2D",
     "DownBlock2D"
   ],
@@ -59,8 +63,10 @@
   "up_block_types": [
     "UpBlock2D",
     "CrossAttnUpBlock2D",
-    "CrossAttnUpBlock2D",
-    "CrossAttnUpBlock2D"
+    "UpBlock2D",
+    "UpBlock2D",
+    "UpBlock2D",
+    "UpBlock2D"
   ],
   "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:476dae627c6a3630fa790f257d2617c4a0703052463ec70f9cf338361d020367
-size 3438144492
+oid sha256:ce72cfc1d976c37ae091e52bad58e19cdf34776eccc05611252c0bc8ab339e7f
+size 574476604