Updating Latest Model

prompt_encoder/config.json

@@ -2,7 +2,7 @@
   "_class_name": "AttributeEmbeddingModel",
   "_diffusers_version": "0.20.2",
   "_name_or_path": "tpremoli/MLD-CelebA-128-80k",
-  "embedding_size": 512,
-  "intermediate_size": 256,
+  "embedding_size": 256,
+  "intermediate_size": 128,
   "num_attributes": 40
 }

prompt_encoder/diffusion_pytorch_model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:42bd946f34c22a94430ec1e2730c516e27b2cd5730d017896bf2a04966e79fed
-size 20976936
+oid sha256:5199cd0ff0738c2f598b6e60f4b8fc93b3d8f988d57f2599e7b76c9a63d5ef50
+size 5245736

unet/config.json

@@ -7,10 +7,10 @@
   "addition_embed_type_num_heads": 64,
   "addition_time_embed_dim": null,
   "attention_head_dim": [
+    64,
     32,
     16,
-    8,
-    4
+    8
   ],
   "attention_type": "default",
   "block_out_channels": [
@@ -24,7 +24,7 @@
   "class_embeddings_concat": false,
   "conv_in_kernel": 3,
   "conv_out_kernel": 3,
-  "cross_attention_dim": 256,
+  "cross_attention_dim": 128,
   "cross_attention_norm": null,
   "down_block_types": [
     "DownBlock2D",

unet/diffusion_pytorch_model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9edfc377a59270c75f26c937663d54d67afce10ea97771ac26b76e0876e54e5e
-size 1149558776
+oid sha256:92c597e39b967690307dd9181712139cac305ca77c3333b6677dc4f9c37535ac
+size 1148641272