fix bucketing doesn't work in controlnet training

This commit is contained in:
Kohya S
2023-08-20 12:24:40 +09:00
parent 2841927dba
commit e191892824
3 changed files with 10 additions and 3 deletions

View File

@@ -113,6 +113,8 @@ def train(args):
assert (
train_dataset_group.is_latent_cacheable()
), "when caching latents, either color_aug or random_crop cannot be used / latentをキャッシュするときはcolor_augとrandom_cropは使えません"
else:
print("WARNING: random_crop is not supported yet for ControlNet training / ControlNetの学習ではrandom_cropはまだサポートされていません")
if args.cache_text_encoder_outputs:
assert (