Training failed with missing values on BrushData

[xduzhangjiayu]

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.py with BrushData

Logs

No response

System Info

Diffusers: 0.31.0
torch: 2.4.0

Who can help?

No response