I trained BrushNet with BrushData, but some values of BrushData (e.g. width, height or segmentation) is missing, so the training is broken in dataloader process, is anyone meet the same issue with me and know how to fix it (skip this data and continue training) in training code, thanks so much.
Describe the bug
I trained BrushNet with BrushData, but some values of BrushData (e.g. width, height or segmentation) is missing, so the training is broken in dataloader process, is anyone meet the same issue with me and know how to fix it (skip this data and continue training) in training code, thanks so much.
Reproduction
Run
train_brushnet.pywith BrushDataLogs
No response
System Info
Diffusers: 0.31.0 torch: 2.4.0
Who can help?
No response