support SD3.5M

This commit is contained in:
Kohya S
2024-10-30 12:51:49 +09:00
parent 75554867ce
commit bdddc20d68
5 changed files with 99 additions and 58 deletions

View File

@@ -239,6 +239,13 @@ def add_sd3_training_arguments(parser: argparse.ArgumentParser):
default=0.0,
help="Dropout rate for T5 encoder, default is 0.0 / T5エンコーダのドロップアウト率、デフォルトは0.0",
)
parser.add_argument(
"--pos_emb_random_crop_rate",
type=float,
default=0.0,
help="Random crop rate for positional embeddings, default is 0.0. Only for SD3.5M"
" / 位置埋め込みのランダムクロップ率、デフォルトは0.0。SD3.5M以外では予期しない動作になります",
)
# copy from Diffusers
parser.add_argument(