Mirrors
/
EveryDream-trainer
mirror of https://github.com/victorchall/EveryDream-trainer.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

tweak yamls

This commit is contained in:
Victor Hall 2022-11-18 23:52:22 -05:00
parent 7098971cd6
commit a8a2e51bc3
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
configs/stable-diffusion/v1-finetune_everydream.yaml
View File

@ -74,7 +74,7 @@ data:
repeats: 5 # rough suggestions: 5 with 5000+ images, 15 for 1000 images, use micro yaml for <100
debug_level: 1 # 1 to print if images are dropped due to multiple-aspect ratio image batching
conditional_dropout: 0.04 # experimental, likelihood to drop the caption, may help with poorly captioned images
resolution: 512 # 512, 576, or 640, increases VRAM substantially
resolution: 512 # use 512 for 24GB, can use 576, 640, 704, 768, on higher VRAM cards only..
validation:
target: ldm.data.ed_validate.EDValidateBatch
params:

6
configs/stable-diffusion/v1-finetune_huge.yaml
View File

@ -71,15 +71,15 @@ data:
train:
target: ldm.data.every_dream.EveryDreamBatch
params:
repeats: 1 # rough suggestions: 5 with 5000+ images, 15 for 1000 images, use micro yaml for <100
repeats: 1 # suggest 1 for 10k+ images
debug_level: 1 # 1 to print if images are dropped due to multiple-aspect ratio image batching
conditional_dropout: 0.04 # experimental, likelihood to drop the caption, may help with poorly captioned images
resolution: 512 # set to 1 or 2 to use larger image sizes for training, USES LOTS OF VRAM! Requires 40GB+
resolution: 512 # use 512 for 24GB, can use 576, 640, 704, 768, on higher VRAM cards only..
seed: -1 # use -1 for random seed, affects ordering of images and shuffling
validation:
target: ldm.data.ed_validate.EDValidateBatch
params:
repeats: 0.2
repeats: 0.2 # suggest low fractions for 10k+ images
test:
target: ldm.data.ed_validate.EDValidateBatch
params: