VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-10 15:00:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix assertion for experimental impl ref #1389

Browse Source
This commit is contained in:
Kohya S
2024-06-26 20:43:31 +09:00
parent 8f2ba27869
commit 828a581e29
1 changed files with 12 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
sd3_train.py
View File

@@ -60,9 +60,19 @@ def train(args):
assert (
not args.weighted_captions
), "weighted_captions is not supported currently / weighted_captionsは現在サポートされていません"
# assert (
# not args.train_text_encoder or not args.cache_text_encoder_outputs
# ), "cache_text_encoder_outputs is not supported when training text encoder / text encoderを学習するときはcache_text_encoder_outputsはサポートされていません"
# training text encoder is not supported
assert (
not args.train_text_encoder or not args.cache_text_encoder_outputs
), "cache_text_encoder_outputs is not supported when training text encoder / text encoderを学習するときはcache_text_encoder_outputsはサポートされていません"
not args.train_text_encoder
), "training text encoder is not supported currently / text encoderの学習は現在サポートされていません"
# training without text encoder cache is not supported
assert (
args.cache_text_encoder_outputs
), "training without text encoder cache is not supported currently / text encoderのキャッシュなしの学習は現在サポートされていません"
# if args.block_lr:
# block_lrs = [float(lr) for lr in args.block_lr.split(",")]