torch>=2.1.0
torchvision>=0.16.0
numpy>=1.24.0
diffusers>=0.25.0
transformers>=4.36.0
Pillow>=10.0.0
einops>=0.7.0
safetensors>=0.4.0
tqdm>=4.60.0
PyYAML>=6.0

[all]
pytorch-image-translation-model[dev,metrics,training]

[dev]
pytest>=7.0.0
ruff>=0.1.0
build
twine

[metrics]
torchmetrics>=1.0.0
lpips>=0.1.4
torch-fidelity>=0.3.0
scipy>=1.10.0

[training]
accelerate>=0.25.0
peft>=0.7.0
datasets>=2.14.0
tensorboard>=2.14.0
swanlab
