From 9a3c9a09e67f58498283256c6b39fefe473ecd95 Mon Sep 17 00:00:00 2001
From: Daniele Acquaviva <24587407+dacquaviva@users.noreply.github.com>
Date: Tue, 29 Nov 2022 21:12:59 +0000
Subject: [PATCH] add style templetes

---
 textual_inversion_training.py | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

diff --git a/textual_inversion_training.py b/textual_inversion_training.py
index 3c6607a..b306c36 100644
--- a/textual_inversion_training.py
+++ b/textual_inversion_training.py
@@ -65,6 +65,28 @@
     "a photo of a small {}",
 ]
 
+style_templates = [
+    "a painting in the style of {}",
+    "a rendering in the style of {}",
+    "a cropped painting in the style of {}",
+    "the painting in the style of {}",
+    "a clean painting in the style of {}",
+    "a dirty painting in the style of {}",
+    "a dark painting in the style of {}",
+    "a picture in the style of {}",
+    "a cool painting in the style of {}",
+    "a close-up painting in the style of {}",
+    "a bright painting in the style of {}",
+    "a cropped painting in the style of {}",
+    "a good painting in the style of {}",
+    "a close-up painting in the style of {}",
+    "a rendition in the style of {}",
+    "a nice painting in the style of {}",
+    "a small painting in the style of {}",
+    "a weird painting in the style of {}",
+    "a large painting in the style of {}",
+]
+
 class TextualInversionDataset:
     def __init__(self, tokenizer, images, learnable_property="object", size=512,
                  repeats=100, interpolation=Resampling.BICUBIC, flip_p=0.5, set="train", placeholder_token="*"):