From f8bfa5f35751e247a79ba15411fe2dabf79ffa3a Mon Sep 17 00:00:00 2001
From: harubaru <seizou.genkei@protonmail.com>
Date: Wed, 12 Oct 2022 18:32:51 -0700
Subject: [PATCH] fix config

---
 configs/stable-diffusion/v1-4-finetune-test.yaml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/configs/stable-diffusion/v1-4-finetune-test.yaml b/configs/stable-diffusion/v1-4-finetune-test.yaml
index d2bb03e..eddbe34 100644
--- a/configs/stable-diffusion/v1-4-finetune-test.yaml
+++ b/configs/stable-diffusion/v1-4-finetune-test.yaml
@@ -77,13 +77,13 @@ data:
     num_workers: 4
     wrap: false
     train:
-      target: ldm.data.local.LocalBase
+      target: ldm.data.local.LocalDanbooruBase
       params:
         size: 512
         mode: "train"
         ucg: 0.1 # unconditional guidance training
     validation:
-      target: ldm.data.local.LocalBase
+      target: ldm.data.local.LocalDanbooruBase
       params:
         size: 512
         mode: "val"