diff --git a/train_configs/training_args_style_sd15.json b/train_configs/training_args_style_sd15.json
index 7c0e4b6..e82a307 100644
--- a/train_configs/training_args_style_sd15.json
+++ b/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,
diff --git a/trainer/preprocess.py b/trainer/preprocess.py
index 213a8f2..2425250 100755
--- a/trainer/preprocess.py
+++ b/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