Issues running

Hi, Thank you for your great work.

When I try to run training code
```python
!accelerate launch --mixed_precision fp16 train_t2i_custom_v2.py --config=configs/custom.py\
```

I receive the following error:
```
TypeError: __init__() got an unexpected keyword argument 'debug'
```

Thanks in advance :)