search

crowsonkb / k-diffusion

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

Move training-only dependencies to [train] extra #60

Open akx opened 1 year ago

akx commented 1 year ago

This PR moves the dependencies required only for training to a [train] package extra, to make it lighter for downstream consumers of the package (in line with the readme's statement "only library code that others can depend on").

Due to how the project's __init__.py imports every submodule, I had to move the respective imports to be quite late.

akx commented 10 months ago

@crowsonkb Rebased on top of 0.1.0.

akx commented 9 months ago

@crowsonkb Rebased on top of 0.1.1.

akx commented 5 months ago

@crowsonkb Rebased on top of 0.2.0.