StableDiffusion/KohyaSS
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix issue 335

Browse Source
This commit is contained in:
bmaltais 2023-03-08 19:33:53 -05:00
parent 54a7537fdf
commit 532552650b
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
library/common_gui.py
View File

@ -790,7 +790,7 @@ def run_cmd_advanced_training(**kwargs):
f' --keep_tokens="{kwargs.get("keep_tokens", "")}"' f' --keep_tokens="{kwargs.get("keep_tokens", "")}"'
if int(kwargs.get('keep_tokens', 0)) > 0 if int(kwargs.get('keep_tokens', 0)) > 0
else '', else '',
f' --caption_dropout_every_n_epochs="{kwargs.get("caption_dropout_every_n_epochs", "")}"' f' --caption_dropout_every_n_epochs="{int(kwargs.get("caption_dropout_every_n_epochs", 0))}"'
if int(kwargs.get('caption_dropout_every_n_epochs', 0)) > 0 if int(kwargs.get('caption_dropout_every_n_epochs', 0)) > 0
else '', else '',
f' --caption_dropout_rate="{kwargs.get("caption_dropout_rate", "")}"' f' --caption_dropout_rate="{kwargs.get("caption_dropout_rate", "")}"'