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