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-09-22 17:44:19 -07:00 committed by GitHub
parent c5f2775beb
commit 5b169173d5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
configs/stable-diffusion/v1-finetune-danboorubase-8gpu.yaml
View File

@ -71,17 +71,17 @@ model:
data:
target: main.DataModuleFromConfig
params:
batch_size: 4
num_workers: 4
batch_size: 1
num_workers: 1
wrap: false
train:
target: ldm.data.local.LocalDanbooruBase
target: ldm.data.localdanboorubase.LocalDanbooruBase
params:
data_root: "./dataset"
size: 768
mode: "train"
validation:
target: ldm.data.local.LocalDanbooruBase
target: ldm.data.localdanboorubase.LocalDanbooruBase
params:
data_root: "./dataset"
size: 768