1 files changed, 1 insertions, 1 deletions

diff --git a/dreambooth.py b/dreambooth.py
index 0c5c42a..0e69d79 100644
--- a/dreambooth.py
+++ b/dreambooth.py
@@ -23,7 +23,7 @@ from slugify import slugify
 from pipelines.stable_diffusion.vlpn_stable_diffusion import VlpnStableDiffusion
 import json
 
-from data.dreambooth.csv import CSVDataModule
+from data.csv import CSVDataModule
 
 logger = get_logger(__name__)