fix config

2022-10-12 18:32:51 -07:00 · 2022-10-12 18:32:51 -07:00 · f8bfa5f357
parent 169b63e179
commit f8bfa5f357
1 changed files with 2 additions and 2 deletions
--- 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"