Updating Latest Model

prompt_encoder/config.json

@@ -1,7 +1,7 @@
 {
   "_class_name": "AttributeEmbeddingModel",
   "_diffusers_version": "0.20.2",
-  "embedding_size": 40,
-  "intermediate_size": 40,
+  "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:d6bce4cf92e2b62842b4485be8dd225a2442fe9cdcc8750e2a20e1a410197a65
-size 256768
+oid sha256:271c28a29f32af47c166a711accd7383970ff5cacf115a4d073f8468e4c17ff6
+size 5245736

samples/prompts-002.txt

@@ -0,0 +1,9 @@
+['NO 5_o_Clock_Shadow', 'NO Arched_Eyebrows', 'YES Attractive', 'NO Bags_Under_Eyes', 'NO Bald', 'NO Bangs', 'NO Big_Lips', 'NO Big_Nose', 'NO Black_Hair', 'NO Blond_Hair', 'NO Blurry', 'YES Brown_Hair', 'NO Bushy_Eyebrows', 'NO Chubby', 'NO Double_Chin', 'NO Eyeglasses', 'NO Goatee', 'NO Gray_Hair', 'YES Heavy_Makeup', 'YES High_Cheekbones', 'NO Male', 'YES Mouth_Slightly_Open', 'NO Mustache', 'NO Narrow_Eyes', 'YES No_Beard', 'YES Oval_Face', 'NO Pale_Skin', 'NO Pointy_Nose', 'NO Receding_Hairline', 'NO Rosy_Cheeks', 'NO Sideburns', 'YES Smiling', 'NO Straight_Hair', 'YES Wavy_Hair', 'NO Wearing_Earrings', 'NO Wearing_Hat', 'YES Wearing_Lipstick', 'NO Wearing_Necklace', 'NO Wearing_Necktie', 'YES Young']
+['YES 5_o_Clock_Shadow', 'NO Arched_Eyebrows', 'YES Attractive', 'NO Bags_Under_Eyes', 'NO Bald', 'NO Bangs', 'YES Big_Lips', 'NO Big_Nose', 'YES Black_Hair', 'NO Blond_Hair', 'NO Blurry', 'NO Brown_Hair', 'YES Bushy_Eyebrows', 'NO Chubby', 'NO Double_Chin', 'NO Eyeglasses', 'NO Goatee', 'NO Gray_Hair', 'NO Heavy_Makeup', 'NO High_Cheekbones', 'YES Male', 'NO Mouth_Slightly_Open', 'NO Mustache', 'NO Narrow_Eyes', 'NO No_Beard', 'NO Oval_Face', 'NO Pale_Skin', 'YES Pointy_Nose', 'NO Receding_Hairline', 'NO Rosy_Cheeks', 'NO Sideburns', 'NO Smiling', 'NO Straight_Hair', 'NO Wavy_Hair', 'NO Wearing_Earrings', 'NO Wearing_Hat', 'NO Wearing_Lipstick', 'NO Wearing_Necklace', 'NO Wearing_Necktie', 'YES Young']
+['YES 5_o_Clock_Shadow', 'YES Arched_Eyebrows', 'NO Attractive', 'NO Bags_Under_Eyes', 'NO Bald', 'NO Bangs', 'YES Big_Lips', 'NO Big_Nose', 'YES Black_Hair', 'NO Blond_Hair', 'NO Blurry', 'NO Brown_Hair', 'YES Bushy_Eyebrows', 'NO Chubby', 'NO Double_Chin', 'NO Eyeglasses', 'NO Goatee', 'NO Gray_Hair', 'NO Heavy_Makeup', 'NO High_Cheekbones', 'YES Male', 'YES Mouth_Slightly_Open', 'NO Mustache', 'NO Narrow_Eyes', 'NO No_Beard', 'NO Oval_Face', 'NO Pale_Skin', 'NO Pointy_Nose', 'NO Receding_Hairline', 'NO Rosy_Cheeks', 'NO Sideburns', 'NO Smiling', 'NO Straight_Hair', 'NO Wavy_Hair', 'NO Wearing_Earrings', 'NO Wearing_Hat', 'NO Wearing_Lipstick', 'NO Wearing_Necklace', 'NO Wearing_Necktie', 'YES Young']
+['NO 5_o_Clock_Shadow', 'NO Arched_Eyebrows', 'NO Attractive', 'NO Bags_Under_Eyes', 'NO Bald', 'NO Bangs', 'NO Big_Lips', 'NO Big_Nose', 'NO Black_Hair', 'NO Blond_Hair', 'YES Blurry', 'NO Brown_Hair', 'NO Bushy_Eyebrows', 'NO Chubby', 'NO Double_Chin', 'NO Eyeglasses', 'NO Goatee', 'NO Gray_Hair', 'NO Heavy_Makeup', 'NO High_Cheekbones', 'NO Male', 'NO Mouth_Slightly_Open', 'NO Mustache', 'NO Narrow_Eyes', 'YES No_Beard', 'NO Oval_Face', 'NO Pale_Skin', 'NO Pointy_Nose', 'NO Receding_Hairline', 'NO Rosy_Cheeks', 'NO Sideburns', 'NO Smiling', 'NO Straight_Hair', 'NO Wavy_Hair', 'YES Wearing_Earrings', 'NO Wearing_Hat', 'NO Wearing_Lipstick', 'NO Wearing_Necklace', 'NO Wearing_Necktie', 'YES Young']
+['NO 5_o_Clock_Shadow', 'NO Arched_Eyebrows', 'YES Attractive', 'NO Bags_Under_Eyes', 'NO Bald', 'YES Bangs', 'NO Big_Lips', 'NO Big_Nose', 'YES Black_Hair', 'NO Blond_Hair', 'NO Blurry', 'NO Brown_Hair', 'NO Bushy_Eyebrows', 'NO Chubby', 'NO Double_Chin', 'NO Eyeglasses', 'NO Goatee', 'NO Gray_Hair', 'YES Heavy_Makeup', 'NO High_Cheekbones', 'NO Male', 'NO Mouth_Slightly_Open', 'NO Mustache', 'NO Narrow_Eyes', 'YES No_Beard', 'YES Oval_Face', 'NO Pale_Skin', 'NO Pointy_Nose', 'NO Receding_Hairline', 'NO Rosy_Cheeks', 'NO Sideburns', 'YES Smiling', 'YES Straight_Hair', 'NO Wavy_Hair', 'NO Wearing_Earrings', 'NO Wearing_Hat', 'YES Wearing_Lipstick', 'NO Wearing_Necklace', 'NO Wearing_Necktie', 'YES Young']
+['YES 5_o_Clock_Shadow', 'NO Arched_Eyebrows', 'YES Attractive', 'NO Bags_Under_Eyes', 'NO Bald', 'NO Bangs', 'NO Big_Lips', 'YES Big_Nose', 'YES Black_Hair', 'NO Blond_Hair', 'NO Blurry', 'NO Brown_Hair', 'NO Bushy_Eyebrows', 'NO Chubby', 'NO Double_Chin', 'YES Eyeglasses', 'YES Goatee', 'NO Gray_Hair', 'NO Heavy_Makeup', 'NO High_Cheekbones', 'YES Male', 'YES Mouth_Slightly_Open', 'YES Mustache', 'NO Narrow_Eyes', 'NO No_Beard', 'NO Oval_Face', 'NO Pale_Skin', 'YES Pointy_Nose', 'NO Receding_Hairline', 'NO Rosy_Cheeks', 'NO Sideburns', 'NO Smiling', 'NO Straight_Hair', 'YES Wavy_Hair', 'NO Wearing_Earrings', 'NO Wearing_Hat', 'NO Wearing_Lipstick', 'NO Wearing_Necklace', 'NO Wearing_Necktie', 'YES Young']
+['NO 5_o_Clock_Shadow', 'NO Arched_Eyebrows', 'NO Attractive', 'YES Bags_Under_Eyes', 'NO Bald', 'NO Bangs', 'YES Big_Lips', 'NO Big_Nose', 'YES Black_Hair', 'NO Blond_Hair', 'NO Blurry', 'NO Brown_Hair', 'NO Bushy_Eyebrows', 'NO Chubby', 'NO Double_Chin', 'NO Eyeglasses', 'NO Goatee', 'NO Gray_Hair', 'NO Heavy_Makeup', 'YES High_Cheekbones', 'YES Male', 'YES Mouth_Slightly_Open', 'YES Mustache', 'NO Narrow_Eyes', 'NO No_Beard', 'NO Oval_Face', 'NO Pale_Skin', 'NO Pointy_Nose', 'NO Receding_Hairline', 'NO Rosy_Cheeks', 'NO Sideburns', 'YES Smiling', 'YES Straight_Hair', 'NO Wavy_Hair', 'NO Wearing_Earrings', 'NO Wearing_Hat', 'NO Wearing_Lipstick', 'NO Wearing_Necklace', 'NO Wearing_Necktie', 'YES Young']
+['NO 5_o_Clock_Shadow', 'NO Arched_Eyebrows', 'NO Attractive', 'YES Bags_Under_Eyes', 'NO Bald', 'NO Bangs', 'NO Big_Lips', 'YES Big_Nose', 'NO Black_Hair', 'NO Blond_Hair', 'NO Blurry', 'YES Brown_Hair', 'YES Bushy_Eyebrows', 'NO Chubby', 'NO Double_Chin', 'NO Eyeglasses', 'NO Goatee', 'NO Gray_Hair', 'NO Heavy_Makeup', 'YES High_Cheekbones', 'NO Male', 'NO Mouth_Slightly_Open', 'NO Mustache', 'NO Narrow_Eyes', 'YES No_Beard', 'NO Oval_Face', 'NO Pale_Skin', 'NO Pointy_Nose', 'NO Receding_Hairline', 'NO Rosy_Cheeks', 'NO Sideburns', 'YES Smiling', 'NO Straight_Hair', 'NO Wavy_Hair', 'YES Wearing_Earrings', 'NO Wearing_Hat', 'NO Wearing_Lipstick', 'NO Wearing_Necklace', 'NO Wearing_Necktie', 'YES Young']
+['NO 5_o_Clock_Shadow', 'YES Arched_Eyebrows', 'NO Attractive', 'NO Bags_Under_Eyes', 'NO Bald', 'NO Bangs', 'NO Big_Lips', 'NO Big_Nose', 'NO Black_Hair', 'YES Blond_Hair', 'NO Blurry', 'NO Brown_Hair', 'NO Bushy_Eyebrows', 'NO Chubby', 'NO Double_Chin', 'NO Eyeglasses', 'NO Goatee', 'NO Gray_Hair', 'NO Heavy_Makeup', 'NO High_Cheekbones', 'NO Male', 'YES Mouth_Slightly_Open', 'NO Mustache', 'YES Narrow_Eyes', 'YES No_Beard', 'NO Oval_Face', 'NO Pale_Skin', 'NO Pointy_Nose', 'YES Receding_Hairline', 'NO Rosy_Cheeks', 'NO Sideburns', 'YES Smiling', 'YES Straight_Hair', 'NO Wavy_Hair', 'YES Wearing_Earrings', 'NO Wearing_Hat', 'YES Wearing_Lipstick', 'YES Wearing_Necklace', 'NO Wearing_Necktie', 'NO Young']

unet/config.json

@@ -24,7 +24,7 @@
   "class_embeddings_concat": false,
   "conv_in_kernel": 3,
   "conv_out_kernel": 3,
-  "cross_attention_dim": 40,
+  "cross_attention_dim": 128,
   "cross_attention_norm": null,
   "down_block_types": [
     "CrossAttnDownBlock2D",

unet/diffusion_pytorch_model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:cd2f5cb550221f1056a01c067b22c791dd1a946aa6017a9764999bf25d5e1c9e
-size 3365483808
+oid sha256:dfaa8cf7244ac4fc63a2f6ba2a5f8b06716aef7ca4c6bf44abd109a902dae58a
+size 3374269768