Training failed with missing values on BrushData
Opened this issue · 0 comments
xduzhangjiayu commented
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