From 9cb2d6bbf887ca5a669c9ab16e16c90cd0b1e03f Mon Sep 17 00:00:00 2001
From: Li Wenyun <liwy@pcl.ac.cn>
Date: Sun, 3 Dec 2023 11:34:01 +0800
Subject: [PATCH] add attribute

---
 config/config.yaml | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/config/config.yaml b/config/config.yaml
index 1182e25..8c7583c 100644
--- a/config/config.yaml
+++ b/config/config.yaml
@@ -13,6 +13,11 @@ paths:
   classifier: checkpoint/
   stylegan: pretrained_models/stylegan2-ffhq-config-f.pt
 
+prompt: red lipstick
+# available attributes
+# ['Blond_Hair', 'Wavy_Hair', 'Young', 'Eyeglasses', 'Heavy_Makeup', 'Rosy_Cheeks',
+#                               'Chubby', 'Mouth_Slightly_Open', 'Bushy_Eyebrows', 'Wearing_Lipstick', 'Smiling',
+#                               'Arched_Eyebrows', 'Bangs', 'Wearing_Earrings', 'Bags_Under_Eyes', 'Receding_Hairline', 'Pale_Skin']
 
 optim:
   batch_size: 8