search

crowsonkb / k-diffusion

Karras et al. (2022) diffusion models for PyTorch
MIT License
2.21k stars 371 forks source link

Use torchsde-brownian instead of full torchsde #82

Closed akx closed 8 months ago

akx commented 10 months ago

This PR switches the dependency on torchsde to my torchsde-brownian fork.