Accelerate is not working fine

crowsonkb / k-diffusion

Karras et al. (2022) diffusion models for PyTorch

MIT License

2.21k stars 371 forks source link

Accelerate is not working fine #57

Open frangam opened 1 year ago

frangam commented 1 year ago

I think there is a bug using "accelerate launch" for launching train.py with acceleration. If you do not use "accelerate" FID is close to 2 points, but if you use "accelerate" FID reports very high values.... Please, could you fix it?