1 files changed, 1 insertions, 1 deletions

diff --git a/train_lora.py b/train_lora.py
index e213e3d..6e72376 100644
--- a/train_lora.py
+++ b/train_lora.py
@@ -15,7 +15,7 @@ from slugify import slugify
 from diffusers.loaders import AttnProcsLayers
 from diffusers.models.cross_attention import LoRAXFormersCrossAttnProcessor, LoRACrossAttnProcessor
 
-from util import load_config, load_embeddings_from_dir
+from util.files import load_config, load_embeddings_from_dir
 from data.csv import VlpnDataModule, keyword_filter
 from training.functional import train, get_models
 from training.lr import plot_metrics