Problem with stable diffusion training

[pinxuezhao]

Hello! Thanks for your cool repo!
When I tried stable diffusion training with
python3 main.py --logdir /tmp -t --postfix test -b configs/train_colossalai_cifar10.yaml
I found it failed just after starting the training loop:



I' m not sure if there are any details that I did wrong.
Any suggestions will be highly appreciated!

[JThh]

Try python3 main.py --logdir /tmp -t --postfix test -b configs/train_colossalai_cifar10.yaml --placement_policy cuda

It was likely due to a previous version of defaulting to auto placement, which often introduced tensor device errors.

[binmakeswell]

We have updated a lot. This issue was closed due to inactivity. Thanks.