fix: textual_inversion

textual_inversion.py

@@ -25,7 +25,8 @@ from diffusers import (
 )
 from diffusers.optimization import get_scheduler
 from diffusers.pipelines.stable_diffusion import StableDiffusionSafetyChecker
-from diffusers.utils import check_min_version
+
+# from diffusers.utils import check_min_version
 from huggingface_hub import HfFolder, Repository, whoami
 
 # TODO: remove and import from diffusers.utils when the new version of diffusers is released
@@ -56,7 +57,7 @@ else:
 
 
 # Will error if the minimal version of diffusers is not installed. Remove at your own risks.
-check_min_version("0.10.0.dev0")
+# check_min_version("0.10.0.dev0")
 
 
 logger = get_logger(__name__)
@@ -316,25 +317,25 @@ imagenet_templates_small = [
 ]
 
 imagenet_style_templates_small = [
-    "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 {}",
+    "a painting of {}, art by *",
+    "a rendering of {}, art by *",
+    "a cropped painting of {}, art by *",
+    "the painting of {}, art by *",
+    "a clean painting of {}, art by *",
+    "a dirty painting of {}, art by *",
+    "a dark painting of {}, art by *",
+    "a picture of {}, art by *",
+    "a cool painting of {}, art by *",
+    "a close-up painting of {}, art by *",
+    "a bright painting of {}, art by *",
+    "a cropped painting of {}, art by *",
+    "a good painting of {}, art by *",
+    "a close-up painting of {}, art by *",
+    "a rendition of {}, art by *",
+    "a nice painting of {}, art by *",
+    "a small painting of {}, art by *",
+    "a weird painting of {}, art by *",
+    "a large painting of {}, art by *",
 ]
 
 
@@ -392,7 +393,7 @@ class TextualInversionDataset(Dataset):
         example = {}
         image = Image.open(self.image_paths[i % self.num_images])
 
-        if not image.mode == "RGB":
+        if image.mode != "RGB":
             image = image.convert("RGB")
 
         placeholder_string = self.placeholder_token
@@ -449,13 +450,11 @@ def freeze_params(params):
 
 def main():
     args = parse_args()
-    logging_dir = os.path.join(args.output_dir, args.logging_dir)
+    # logging_dir = os.path.join(args.output_dir, args.logging_dir)
 
     accelerator = Accelerator(
         gradient_accumulation_steps=args.gradient_accumulation_steps,
         mixed_precision=args.mixed_precision,
-        log_with="tensorboard",
-        logging_dir=logging_dir,
     )
 
     # If passed along, set the training seed now.