From 515f0f1fdc9a76bf63bd746c291dcfec7fc747fb Mon Sep 17 00:00:00 2001
From: Volpeon <git@volpeon.ink>
Date: Thu, 13 Oct 2022 21:11:53 +0200
Subject: Added support for Aesthetic Gradients

---
 data/csv.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'data/csv.py')

diff --git a/data/csv.py b/data/csv.py
index 253ce9e..aad970c 100644
--- a/data/csv.py
+++ b/data/csv.py
@@ -23,7 +23,7 @@ class CSVDataModule(pl.LightningDataModule):
             tokenizer,
             instance_identifier,
             class_identifier=None,
-            class_subdir="db_cls",
+            class_subdir="cls",
             num_class_images=100,
             size=512,
             repeats=100,
-- 
cgit v1.2.3-54-g00ecf