ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-01-26 16:39:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

added target weight to targets

Browse Source
This commit is contained in:
Jaret Burkett
2023-07-23 14:08:37 -06:00
parent 452f2a6da2
commit 9a2819900c
2 changed files with 17 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
config/examples/train_slider.example.yml
View File

@@ -158,9 +158,10 @@ config:
# negative is the prompt for the negative side of the slider and works the same as positive
# it does not necessarily work the same as a negative prompt when generating images
negative: "dog"
# LoRA weight to train this target. I recommend 1.0. Just leave it, it won't work
# how you expect if you change it
multiplier: 1.0
# the loss for this target is multiplied by this number.
# if you are doing more than one target it may be good to set less important ones
# to a lower number like 0.1 so they dont outweigh the primary target
weight: 1.0
# You can put any information you want here, and it will be saved in the model.
# The below is an example, but you can put your grocery list in it if you want.