AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

Fix kontext finetune issue when batch size >1 (#11921)

Browse Source 
set drop_last to True

Signed-off-by: mymusise <mymusise1@gmail.com>
This commit is contained in:
Chengxi Guo
2025-07-19 07:38:58 +08:00
committed by GitHub
parent 5dc503aa28
commit cde02b061b
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/dreambooth/train_dreambooth_lora_flux_kontext.py
View File

@@ -1614,7 +1614,7 @@ def main(args):
)
if args.cond_image_column is not None:
logger.info("I2I fine-tuning enabled.")
batch_sampler = BucketBatchSampler(train_dataset, batch_size=args.train_batch_size, drop_last=False)
batch_sampler = BucketBatchSampler(train_dataset, batch_size=args.train_batch_size, drop_last=True)
train_dataloader = torch.utils.data.DataLoader(
train_dataset,
batch_sampler=batch_sampler,