training a picture a style

[EnigmaHong]

accelerate launch --num_processes 8 --mixed_precision fp16 train_t2i_custom_v2.py --config=configs/custom.py

I use the above command to train a picture style model, loss is always shock, grad_norm: nan (nan), the effect is not good, may I ask what is the problem?