1 files changed, 1 insertions, 1 deletions

diff --git a/train_ti.py b/train_ti.py
index c139cc0..b9d6e56 100644
--- a/train_ti.py
+++ b/train_ti.py
@@ -12,7 +12,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, add_placeholder_tokens, get_models
 from training.lr import plot_metrics