1 files changed, 1 insertions, 1 deletions

diff --git a/train_dreambooth.py b/train_dreambooth.py
index 9b91172..dd2bf6e 100644
--- a/train_dreambooth.py
+++ b/train_dreambooth.py
@@ -13,7 +13,7 @@ from accelerate.logging import get_logger
 from accelerate.utils import LoggerType, set_seed
 from slugify import slugify
 
-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