update preprocess bug

train_configs/training_args_style_sd15.json

@@ -1,7 +1,7 @@
 {
     "name": "twisting_realities_sd15",
     "sd_model_version": "sd15",
-    "lora_training_urls": "https://edenartlab-lfs.s3.amazonaws.com/datasets/twisting_realities.zip",
+    "lora_training_urls": "/home/rednax/Documents/datasets/good_styles/visionary_painting_ygor_marotta_clean",
     "concept_mode": "style",
     "sample_imgs_lora_scale": 0.8,
     "seed": 0,

trainer/preprocess.py

@@ -513,7 +513,10 @@ def florence_caption_dataset(images, captions):
         if not str(filename).endswith("modeling_florence2.py"):
             return get_imports(filename)
         imports = get_imports(filename)
-        imports.remove("flash_attn")
+        try:
+            imports.remove("flash_attn")
+        except:
+            pass
         return imports
     
     with patch("transformers.dynamic_module_utils.get_imports", fixed_get_imports): #workaround for unnecessary flash_attn requirement