Mirrors
/
waifu-diffusion
mirror of https://github.com/harubaru/waifu-diffusion.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix config

This commit is contained in:
harubaru 2022-10-12 18:32:51 -07:00
parent 169b63e179
commit f8bfa5f357
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
configs/stable-diffusion/v1-4-finetune-test.yaml
View File

@ -77,13 +77,13 @@ data:
num_workers: 4 num_workers: 4
wrap: false wrap: false
train: train:
target: ldm.data.local.LocalBase target: ldm.data.local.LocalDanbooruBase
params: params:
size: 512 size: 512
mode: "train" mode: "train"
ucg: 0.1 # unconditional guidance training ucg: 0.1 # unconditional guidance training
validation: validation:
target: ldm.data.local.LocalBase target: ldm.data.local.LocalDanbooruBase
params: params:
size: 512 size: 512
mode: "val" mode: "val"