PyPI page
Home page
Author:
None
Summary:
Aspect ratio bucketing toolkit for diffusion model training. PyTorch native, DDP correct.
Latest version:
0.1.0a3
Required dependencies:
numpy
|
pillow
|
pydantic
|
tomli
|
typer
Optional dependencies:
datasets
|
diffusers
|
matplotlib
|
mypy
|
pyarrow
|
pytest
|
pytest-cov
|
ruff
|
safetensors
|
torch
|
types-pillow
Downloads last day:
6
Downloads last week:
105
Downloads last month:
370