HuggingFace Datasets support with new config

crowsonkb / k-diffusion

Karras et al. (2022) diffusion models for PyTorch

MIT License

2.26k stars 372 forks source link

HuggingFace Datasets support with new config #11

Closed tmabraham closed 2 years ago

tmabraham commented 2 years ago

Essentially the same as my previous PR #5, but using the new dataset config that you added. Also added an example configuration and updated the .gitignore.

crowsonkb commented 2 years ago

I changed some small things then merged this PR: https://github.com/crowsonkb/k-diffusion/commit/caefaadf954e8e5818cd215e74d880e09fd278d9

Thank you! :)